torch.autograd.Function.jvp

static Function.jvp(ctx, *grad_inputs)

定义一个公式，用于使用前向模式自动微分来计算操作的导数。

此函数需要被所有子类重写。它必须接受一个上下文 ctx 作为第一个参数，然后是与 forward() 方法接收到的输入数量相同的参数（对于前向函数中非张量类型的输入将传递 None）。它应该返回与 forward() 输出相同数量的张量。每个参数是相对于给定输入的梯度，每个返回值应该是相对于相应输出的梯度。如果某个输出不是张量或函数对该输出不可微，则可以为该输入传递 None 作为梯度。

你可以使用ctx对象将任何值从前进方向传递到此函数。

返回类型

Any