torch.autograd.Function.jvp¶

static Function.jvp(ctx, *grad_inputs)[原始碼]¶

定義使用正向模式自動微分來微分運算的公式。

此函式應由所有子類別覆寫。它必須接受一個上下文 ctx 作為第一個引數，後跟與 forward() 取得的一樣多的輸入（對於 forward 函式的非 Tensor 輸入，將傳入 None），並且它應該傳回與 forward() 的輸出一樣多的 Tensor。每個引數都是 w.r.t 給定輸入的梯度，並且每個傳回值都應該是 w.r.t. 相對應輸出的梯度。如果輸出不是 Tensor，或者該函式對於該輸出不可微分，您可以只為該輸入傳遞 None 作為梯度。