Bootstrap

【python】pytorch中的detach是什么功能

detach的方法,将variable参数从网络中隔离开,不参与参数更新。

如下:


# y=A(x), z=B(y) 求B中参数的梯度,不求A中参数的梯度
y = A(x)
z = B(y.detach())
z.backward()

 

;