int类型数据转为 torch.cuda.FloatTensor类型数据

2022年12月25日07:59:48

数据类型简介:

数据类型包括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()

 

  • 作者:渣渣崔
  • 原文链接:https://blog.csdn.net/weixin_39663060/article/details/108655508
    更新时间:2022年12月25日07:59:48 ,共 641 字。