\u200E

[ 组合替代实现 ]torch.Tensor.requires_grad_

torch.Tensor.requires_grad_

          torch.Tensor.requires_grad_(requires_grad=True)

         

paddle.Tensor.stop_gradient 

          paddle.Tensor.stop_gradient = False

         

两者功能一致，torch 为 function 调用方式，paddle 为 attribution 赋值方式，具体如下：

参数映射

PyTorch	PaddlePaddle	备注
requires_grad	-	是否计算梯度，Paddle 无此参数，需要转写。

转写示例

requires_grad：是否计算梯度

           # 当 requires_grad 为‘True’时，torch 写法
x.requires_grad_(requires_grad=True)

# paddle 写法
x.stop_gradient = False

# 当 requires_grad 为‘False’时，torch 写法
x.requires_grad_(requires_grad=False)

# paddle 写法
x.stop_gradient = True

          