[ torch 参数更多 ] torch.Tensor.backward¶
torch.Tensor.backward¶
torch.Tensor.backward(gradient=None, retain_graph=None, create_graph=False, inputs=None)
paddle.Tensor.backward¶
paddle.Tensor.backward(grad_tensor=None, retain_graph=False)
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
gradient | grad_tensor | 当前 Tensor 的初始梯度值。仅参数名不一致。 |
retain_graph | retain_graph | 是否保留计算图。 |
create_graph | - | 是否创建梯度图,Paddle 无此参数,暂无转写方式。 |
inputs | - | 计算的起始输入 tensor,Paddle 无此参数,暂无转写方式。 |