torch.autograd.Function.jvp#

static Function.jvp(ctx, *grad_inputs)[source]#

定义使用前向模式自动微分来区分操作的公式。

此函数应由所有子类覆盖。它必须接受一个上下文 ctx 作为第一个参数，后跟 forward() 接收的输入数量（对于 forward 函数的非 tensor 输入将传递 None），并且它应该返回与 forward() 的输出数量相同的张量。每个参数是相对于给定输入的梯度，每个返回值应该是相对于相应输出的梯度。如果某个输出不是 Tensor，或者函数相对于该输出不可微分，则可以将 None 作为该输入的梯度传递。