\u200E

[ 无参数 ]torch.is_inference¶

torch.is_inference ¶

          torch.is_inference(input)

         

paddle.Tensor.stop_gradient ¶

          paddle.Tensor.stop_gradient

         

两者功能一致，无参数。 is_inference 会强制关闭梯度记录。并且不能在中途设置梯度，而 stop_gradient 仅为停止计算该算子梯度，可在中途重新设为 True ，Paddle 为近似实现。

转写示例¶

          # PyTorch 写法
torch.is_inference(x)

# Paddle 写法
x.stop_gradient