torch.tensor与torch.Tensor的区别。
torch.tensor只能从指定的数据创建,但是可以指定数据属性,是否可微分等属性。
import torch
a = [1, 2, 3]
b = torch.tensor(a, requires_grad=True, dtype=torch.float64)
torch.Tensor有一下几种变形:
整数:
torch.ShorTensor 16位
torch.IntTensor 32位
torch.LongTensor 64位
浮点:
torch.FloatTensor=torch.Tensor 32位
torch.DoubleTensor 64位
注意:torch.Tensor(int1, int2,int3)会创建[int1, int2,int3]形状的张量,如果传入可迭代,就会将改可迭代对象变成张量。
import torch
torch.Tensor(3)
'''tensor([-2.6853e+05, 1.9983e-42, 2.3694e-38])'''
torch.Tensor(3, 1)
'''
tensor([[3.2842e-15],
[3.1714e+00],
[2.3694e-38]])
'''
torch.Tensor([3, 1])
'''
tensor([3., 1.])
'''