torch.tensor与torch.Tensor的区别
-
torch.Tensor()的默认数据类型为torch.float32,而torch.tensor()的数据类型为原类型。
-
torch.Tensor()传入数字表示的形状与torch.FloatTensor相同
a = torch.Tensor(1) print(a.type())
-
当torch.Tensor()传入的是一个数字时,返回的不是一个tensor类型的数字,而是一个tensor类型的列表,并且此列表的长度等于传入的数字,列表中的数据类型默认为float32。
b = torch.Tensor(3) print(b)
-
当torch.tensor()传入的是一个数字时,返回的是tensor类型的数字
c = torch.tensor(3) print(c)
-
-
torch.Tensor传入可迭代对象,返回值为tensor类型的列表,且里面的数据默认为float32
d = torch.Tensor([1, 2, 3]) print(d)
-
torch.tensor传入可迭代对象,返回值为tensor类型的列表,且里面的数据的数据类型为输入时可迭代对象里面的数据类型。
e = torch.tensor([2, 2, 3]) print(e)