torch.autograd.Function.jvp¶

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

定义一个公式，用于使用前向模式自动微分来对操作进行微分。

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