1、torch.Tensor()是一种包含单一数据类型元素的多维矩阵,是默认的tensor类型(torch.FlaotTensor
)的简称。
array1=np.arange(12).reshape(3,4) #生成一个列表
array1=torch.Tensor(array1)
print(array1)
2、一个空张量tensor可以通过规定其大小来构建:
torch.IntTensor(2, 4).zero_()
0 0 0 0
0 0 0 0
[torch.IntTensor of size 2x4]
3、修改tensor变量的类型
t1=torch.Tensor([1,2])
t1.int() #修改为int型
t1.long() #修改为long型
t1.double() #修改为double型