本文主要是介绍Pytorch截断计算图:detach操作,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
官方解析:
torch.Tensor.detach():
Returns a new Tensor, detached from the current graph.
The result will never require gradient.
举个例子:
在GAN训练中,训练判别器D时,无需调整生成器G的参数。
fake_image = Generator(noise).detach() # 返回生成图片的Tensor,这个Tensor与生成器G的计算图时分开的。
fake_output = Descriminator(fake_image)
这篇关于Pytorch截断计算图:detach操作的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!