原文:torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda 函数得到。 一般系统默认是torch.FloatTensor类型。例如data torch.Tensor , 是一个 的张量,类型为FloatTensor data.cuda 就转换为GPU的张量类型,torch.cuda.FloatTensor类型。 ...

2020-02-24 15:54 0 5381 推荐指数:

查看详情

torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。 一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量,类型 ...

Tue Dec 22 16:56:00 CST 2020 0 1141
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM