pytorch变量类型转换的介绍和简单示例

1.简单介绍

Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵，与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。我们可以通过使用Type函数来查看变量的类型，一般的系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量，类型为FloatTensor; data.cuda()就转换为GPU的张量类型，torch.cuda.FloatTensor类型。

2.示例

• 张量之间类型的转换

一般只要在Tensor后加long(), int(), double(),float(),byte()等函数就能将Tensor进行类型转换；还可以使用type()函数，data为Tensor数据类型，data.type()为给出data的类型，如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。

例如：Torch.LongTensor--->Torch.FloatTensor, 直接使用data.float()即可，代码如下：

import torch

    data = torch.LongTensor(2, 3)  # 构建Long类型的Tensor
    print(data)
    # 转换为 float类型的Tensor