代码如下:
import torch if __name__ == '__main__': #转换数据类型 #1.创建tensor,默认为torch.FloatTensor a=torch.Tensor(2,3) print(a) #2.使用double()等直接转换数据类型 b=a.double() print(b) #3.使用type()函数转换数据类型 c=a.type(torch.DoubleTensor) print(c) #4.使用type_as()函数 最重要************************** d=a.type_as(b) print(d)