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