[参数完全一致]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。 |