官方解析:
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()函数的作用,它用于从当前计算图中分离出一个新的Tensor,新Tensor不会追踪梯度。文章通过GAN训练实例,说明在训练判别器时如何使用detach()来避免更新生成器的参数。
2772

被折叠的 条评论
为什么被折叠?



