torch.autograd.Function.jvp¶

static Function.jvp(ctx, *grad_inputs)¶

定义一个用于前向模式自动微分的操作的公式。

此函数应由所有子类重写。它必须接受上下文 ctx 作为第一个参数，随后是与 forward() 接收的输入一样多的输入（对于 forward 函数的非张量输入，将传递 None），并且它应返回与 forward() 输出一样多的张量。每个参数是相对于给定输入的梯度，并且每个返回值应是相对于相应输出的梯度。如果某个输出不是张量或函数相对于该输出不可微，您可以简单地为该输入传递 None 作为梯度。