Pytorch多次求导保留计算图的方法

通过调用 backward 我们可以进行一次自动求导,如果我们再调用一次 backward,会发现程序报错,没有办法再做一次。这是因为 PyTorch 默认做完一次自动求导之后,计算图就被丢弃了,所以两次自动求导需要手动设置一个东西,我们通过下面的小例子来说明。

#保留本次自动求导后的计算图
y.backward(retain_graph=True)

猜你喜欢

转载自blog.csdn.net/weixin_41797117/article/details/79913685
今日推荐