[仅参数名不一致]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。