[torch 参数更多]torch.autograd.functional.jvp

torch.autograd.functional.jvp

torch.autograd.functional.jvp(func, inputs, v=None, create_graph=False, strict=False)

paddle.incubate.autograd.jvp

paddle.incubate.autograd.jvp(func, xs, v=None)

PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
func func Python 函数。
inputs xs 函数 func 的输入参数。
v v 用于计算 jvp 的输入向量。
create_graph - 是否创建图,Paddle 无此参数,暂无转写方式。
strict - 是否在存在一个与所有输出无关的输入时抛出错误,Paddle 无此参数,暂无转写方式。