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

数据类型简介:

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值