\u200E

[参数完全一致]torch.autograd.grad_mode.set_grad_enabled

torch.autograd.grad_mode.set_grad_enabled 

          torch.autograd.grad_mode.set_grad_enabled(mode)

         

paddle.set_grad_enabled 

          paddle.set_grad_enabled(mode)

         

两者功能一致，参数完全一致，具体如下：

参数映射

PyTorch	PaddlePaddle	备注
mode	mode	启用（True）或禁用（False）动态图梯度计算。