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.]])

你可能感兴趣的:(Deep-Learning-with-PyTorch-3.7.2 修改存储的值:就地操作)