torch.autograd.Function.jvp

static Function.jvp(ctx, *grad_inputs)

定義使用前向模式自動微分來區分操作的公式。

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

You can use the ctx object to pass any value from the forward to this functions.

返回型別

任何