torch.autograd.Function.jvp¶

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

定义使用前向模式自动微分对操作进行微分公式的公式。

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