在 Pytorch 中,Tensor 和 tensor 都用于生成新的张量,但二者并不相同。
a = torch.Tensor([1, 2])
a # tensor([1., 2.])
a=torch.tensor([1,2])
a # tensor([1, 2])
torch.Tensor
torch.Tensor()
是默认张量类型torch.FloatTensor()
的别名,生成单精度浮点类型的张量。
a=torch.Tensor([1,2])
a.type() # 'torch.FloatTensor'
torch.tensor()
torch.tensor()
的函数原型为
torch.tensor(data, dtype=None, device=None, requires_grad=False)
- data 类型可为:list, tuple, array, scalar 等;
torch.tensor()
可根据原始数据类型生成相应类型的 tensor;
a = torch.tensor([1, 2])
a.type() # 'torch.LongTensor'
a = torch.tensor([1., 2.])
a.type() # 'torch.FloatTensor'
a = np.zeros(2, dtype=np.float64)
a = torch.tensor(a)
a.type() # torch.DoubleTensor