pytorch用法记录2

1.torch.Storage类    使用storage()函数把Tensor数据转换为float类型的Storage数据,再使用tolist() 返回一个包含此存储中元素的列表。 2.detach 计算图截断 detach 的意思是,这个数据和生成它的计算图“脱钩”了,即detach就是截断反向传播的梯度流。GAN中,Train D on fake,G生成的数据会传入D,然后计算loss,再反
相关文章
相关标签/搜索