[仅参数名不一致]torch.autograd.function.FunctionCtx.mark_non_differentiable¶
torch.autograd.function.FunctionCtx.mark_non_differentiable¶
torch.autograd.function.FunctionCtx.mark_non_differentiable(*args)
paddle.autograd.PyLayerContext.mark_non_differentiable¶
paddle.autograd.PyLayerContext.mark_non_differentiable(*tensors)
两者功能一致,仅参数名不一致,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
args | tensors | 需要标记不需要反向的 Tensor。 |