set_grad_enabled¶
创建启用或禁用动态图梯度计算的上下文。
参数¶
mode (bool) - 启用(True)或禁用(False)动态图梯度计算。
返回¶
None
代码示例¶
import paddle
x = paddle.ones([3, 2])
x.stop_gradient = False
with paddle.set_grad_enabled(False):
y = x * 2
with paddle.set_grad_enabled(True):
z = x * 2
print(y.stop_gradient) # True
print(z.stop_gradient) # False