torch.autograd.Function.jvp¶

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

定義一個公式，用於使用前向模式自動微分來對操作進行微分。

所有子類都應覆蓋此函式。它必須接受一個上下文 ctx 作為第一個引數，後跟與 forward() 接收到的輸入數量相同的輸入（對於 forward 函式的非 Tensor 輸入，將傳入 None），並且它應該返回與 forward() 輸出數量相同的 Tensor。每個引數都是對應於給定輸入的梯度，每個返回值都應是對應於相應輸出的梯度。如果輸出不是 Tensor 或函式對於該輸出不可微，您可以直接將 None 作為該輸入的梯度傳入。