标签: tensorflow pytorch
pytorch中的tf.stop_gradient()等效项是什么(提供了一种在反向传播期间不计算某些变量的梯度的方法)?
tf.stop_gradient()
答案 0 :(得分:5)
可以用x.detach()检查吗?
答案 1 :(得分:1)
pytorch中的张量具有requires_grad属性。将其设置为False可以防止对该张量进行梯度计算。
requires_grad
False