首页 > 解决方案 > pytorch 中的 tensorflow stop_gradient 等效项

问题描述

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

标签: tensorflowpytorch

解决方案


你能用 x.detach() 检查一下吗?


推荐阅读