分类：变量转换

2021
07-02

Pytorch 实现变量类型转换

Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵。与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。通过使用Type函数可以查看变量类型。一般系统默认是torch.FloatTensor类型。例如data=torch.Tensor(2,3)是一个2*3的张量，类型为FloatTensor;data.cuda()就转换为GPU的张量类型，torch.cuda.FloatTensor类型。下面简单介绍... 继续阅读 >