torch.Tensor和torch.tensor的区别

 

torch.Tensor和torch.tensor的区别

2019-06-10 16:34:48 Vic_Hao 阅读数 4058更多

分类专栏： Pytorch

在Pytorch中，Tensor和tensor都用于生成新的张量。

>>>  a = torch.Tensor([1, 2])
>>>  a
tensor([1., 2.])
>>> a=torch.tensor([1,2])
>>> a
tensor([1, 2])

• 1
• 2
• 3
• 4
• 5
• 6

首先我们从根源上来看看torch.Tensor()和torch.tensor()区别。

torch.Tensor

torch.Tensor()是Python类，更明确的说，是默认张量类型torch.FloatTensor()的别名，torch.Tensor([1,2]) 会调用Tensor类的构造函数__init__，生成单精度浮点类型的张量。

>>> a=torch.Tensor([1,2])
>>> a.type()
'torch.FloatTensor'

• 1
• 2