torch.autograd.Function.jvp¶

static Function.jvp(ctx, *grad_inputs)[源代码]¶

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

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