Deep-Learning-with-PyTorch-3.7.2 修改存储的值:就地操作

3.7.2 修改存储的值:就地操作

除了上一节中介绍的对张量的操作以外,仅作为Tensor对象的方法存在的操作数量很少。 可以从名称的尾部下划线识别它们,例如zero_,这表示该方法通过修改输入而不是创建新的输出张量并返回它来就地运行。 例如,zero_方法将输入的所有元素清零。 任何不带下划线的方法都将使源张量保持不变,而返回一个新的张量:

# In[73]:
a = torch.ones(3, 2)

# In[74]:
a.zero_()
a

# Out[74]:
tensor([[0., 0.],
        [0., 0.],
        [0., 0.]])
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容