一、张量的数据类型转换

 1.演示data.type(trch.DoubleTensor)

#1.创建张量对象  [6 6 6;6 6 6]
data=torch.full([2,3],6)
print(data.dtype)#默认为torch.int64(LongTensor)
#2.转化为double类型
data=data.type(torch.DoubleTensor)
print(data.dtype)
#3.转换成int类型
data=data.type(torch.IntTensor)
print(data.dtype)
#4.转换成Long类型
data=data.type(torch.LongTensor)
print(data)
print(data.dtype)
#5.转换成float类型
data=data.type(torch.FloatTensor)
print(data)
print(data.dtype)

运行结果：

2.演示data.double()

#1.转化为double类型
data1=data.double()
print(data1.dtype)
data2=data.float()
print(data2.dtype)
print(data2.type())
data3=data.short()
print(data3.dtype)

运行结果：

二、张量的类型转换