Torch detach

2020-01-04  本文已影响0人  winddy_akoky

官网解释:

image.png

即:返回一个tensor变量,且这个变量永远不会有梯度值。这个变量跟原图上的变量共享一块内存,也就说是同一个家伙。
他们俩的in-place都会触发错误。

image.png

这个方法可以用来实现对部分网络求梯度:


image.png
上一篇 下一篇

猜你喜欢

热点阅读