数据类型简介:
数据类型包括Python数据类型、Numpy数据类型和Pytorch中的tensor,Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型。
解决方法:
int是python的数据类型,torch.cuda.FloatTensor是Pytorch的GPU数据类型。
转换方式为:python数据类型->Numpy数据类型->Pytorch的cup tensor->Pytorch的gpu tensor
例如:
a为基本的int类型数据
b=np.array(a), b为numpy数据类型
c=torch.from_numpy(b),c为CPU的tensor
d=c.cuda(),d为GPU的tensor
不同数据类型之间的转化:
python基本数据类型和Numpy数据类型之间的转换:
Numpy--->python 使用data.item(),data为numpy变量
python--->Numpy 使用np.array(data),data为python变量
Tensor与numpy之间的转换:
Tensor---->Numpy 使用 data.numpy(),data为Tensor变量
Numpy ----> Tensor 使用torch.from_numpy(data),data为numpy变量
GPU tensor与CPUtensor之间的转换:
CPU张量 ----> GPU张量, 使用data.cuda()
GPU张量 ----> CPU张量 使用data.cpu()