torch.autograd.Function.jvp¶

static Function.jvp(ctx, *grad_inputs)¶

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

所有子类都必须覆盖此函数。它必须接受一个上下文 ctx 作为第一个参数，后面跟着与 forward() 相同数量的输入（对于前向函数的非张量输入，不会传递任何内容），它应该返回与 forward() 的输出数量相同的张量。每个参数都是关于给定输入的梯度，每个返回值都应该是关于对应输出的梯度。如果输出不是张量，或者函数不能关于该输出进行微分，则可以将 None 作为该输入的梯度传递。