pytorch - PyTorch:new_ones与1

Question

在PyTorch中，new_ones()与ones()有什么区别。例如，

x2.new_ones(3,2, dtype=torch.double)

与

torch.ones(3,2, dtype=torch.double)

Answer 1

为了这个答案，我假设您的x2是先前定义的torch.Tensor。如果然后转到PyTorch documentation，则可以在new_ones()上阅读以下内容：


返回大小为size的张量，并用1填充。默认情况下，
返回的Tensor具有与此相同的torch.dtype和torch.device
张量。


而ones()


返回一个填充有标量值1的张量，其形状为
由变量参数大小定义。


因此，实质上，new_ones允许您在与先前存在的张量（带张量）相同的设备和数据类型上快速创建新的torch.Tensor，而ones()的目的是从头开始创建torch.Tensor（充满）。