tensor中数据类型的相互转换

Pytorch中的Tensor常用的类型转换函数tensor数据类型转换在tensor的后面添加: .int()、.float()、.double()等.同时也可以使用 .to(type) 进行实现。同时也可以使用 type()函数 ,data为Tensor数据类型，data.type()为给出data的类型，如果使用data.type(torch.FloatTensor)则强制转换为torch.

passion-ma

5740人浏览 · 2022-05-07 10:04:56

passion-ma · 2022-05-07 10:04:56 发布

Pytorch中的Tensor常用的类型转换函数

tensor数据类型转换

例如：

a = tensor(282, device='cuda:0')

b = int(a)

print(b) # 282,int类型
    在tensor的后面添加: .int()、.float()、.double()等.
    同时也可以使用 .to(type) 进行实现。
    同时也可以使用 type()函数 ,data为Tensor数据类型，data.type()为给出data的类型，如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。当你不知道要转换为什么类型时，但需要求a1,a2两个张量的乘积，可以使用a1.type_as(a2)将a1转换为a2同类型。
    tensor<–>numpy
    Tensor---->Numpy 使用 tensor.numpy()，tensor为Tensor变量
    Numpy ----> Tensor 使用 torch.from_numpy(data)，data为numpy变量
    tensor<–>Python
    Tensor ----> 单个Python数据，使用data.item()，data为Tensor变量且只能为包含单个数据
    Tensor ----> Python list，使用data.tolist()，data为Tensor变量，返回shape相同的可嵌套的list
    数据存储位置转换
    CPU张量 ----> GPU张量，使用data.cuda()
    GPU张量 ----> CPU张量，使用data.cpu()