【python】torch.tensor 和 Tensor区别

Tensor是python类，tensor是python函数，区别在于默认的数据类型不一样。

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

>>> a.type()
'torch.FloatTensor'

 

>>> a=torch.tensor([1,2])
>>> print(a)
tensor([1, 2])
>>> a.type()
'torch.LongTensor'

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

>>> a=torch.tensor([1,2],dtype=torch.float32)
>>> a.type()
'torch.FloatTensor'