请选择 进入手机版 | 继续访问电脑版
查看: 1386|回复: 0

torch.tensor的类型转换以及和numpy的转换

[复制链接]

665

主题

1234

帖子

6561

积分

xdtech

Rank: 5Rank: 5

积分
6561
发表于 2020-5-14 08:54:11 | 显示全部楼层 |阅读模式
PyTorch中的常用的tensor类型
      PyTorch中的常用的tensor类型包括:
    32位浮点型torch.FloatTensor,
    64位浮点型torch.DoubleTensor,
    16位整型torch.ShortTensor,
    32位整型torch.IntTensor,
    64位整型torch.LongTensor。
类型之间的转换
      一般只要在tensor后加long(), int(), double(),float(),byte()等函数就能将tensor进行类型转换
   
  
      此外,还可以使用type()函数,data为Tensor数据类型,data.type()为给出data的类型,如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。
      a1.type_as(a2)可将a1转换为a2同类型。
tensor和numpy.array转换
  tensor -> numpy.array: data.numpy(),如:
  
  numpy.array -> tensor: torch.from_numpy(data),如:
  
CPU张量和GPU张量之间的转换
  CPU -> GPU: data.cuda()
  GPU -> CPU: data.cpu()



回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表