[参数完全一致]torch.autograd.function.FunctionCtx.save_for_backward
torch.autograd.function.FunctionCtx.save_for_backward
torch.autograd.function.FunctionCtx.save_for_backward(*tensors)
paddle.autograd.PyLayerContext.save_for_backward
paddle.autograd.PyLayerContext.save_for_backward(*tensors)
两者功能一致,参数完全一致,具体如下:
参数映射
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| tensors | tensors | 需要被暂存的 Tensor。 |