pytorch中的tensorflow stop_gradient等效

时间:2018-07-26 02:04:30

标签: tensorflow pytorch

pytorch中的tf.stop_gradient()等效项是什么(提供了一种在反向传播期间不计算某些变量的梯度的方法)?

2 个答案:

答案 0 :(得分:5)

可以用x.detach()检查吗?

答案 1 :(得分:1)

pytorch中的张量具有requires_grad属性。将其设置为False可以防止对该张量进行梯度计算。