【QA篇】Numpy与Pytorch类型转换

在使用Numpy和PyTorch进行数据处理和模型训练时，经常需要进行数据类型的转换。但是，在进行转换时，可能会遇到一些坑，本文将介绍如何解决这些坑。

搬运自Python技术站
链接: 解决Numpy与Pytorch彼此转换时的坑

Numpy与PyTorch的数据类型

在Numpy中，常用的数据类型有int、float、bool等，而在PyTorch中，常用的数据类型有torch.int、torch.float、torch.bool等。这些数据之间的转换需要注意。

Numpy数组转换为PyTorch张量

将Numpy数组转换为PyTorch张量时，需要注意数据类型的转换。下面是一个示例，演示如何将Numpy数组转换为Torch张量。

import numpy as np
import torch

# 创建一个Numpy数组
arr = np.array([1, 2, 3, 4, 5])

# 将Numpy数组转换为PyTorch张量
tensor = torch.from_numpy(arr)

print(tensor)  # tensor([1, 2, 3, 4, 5], dtype=torch.int32)

在上面的示例中，创建了一个Numpy数组arr，然后使用torch.from_numpy函数将其转换为PyTorch张量。
需要注意的是，由于Numpy数组的数据类型是int32，因此转换的PyTorch张量的数据类型也是torch.int32。

PyTorch张量转换为Numpy数组

将PyTorch张量转换为Numpy数组时，同样需要注意数据类型的转换。下面是一个示例，演示如何将PyTorch张量转换为Numpy数组。

import numpy as np
import torch

# 创建一个PyTorch张量
tensor = torch.tensor([1, 2, 3, 4, 5])

# 将PyTorch张量转换为Numpy数组
arr = tensor.numpy()

print(arr)  # [1 2 3 4 5]

在上面的示例中，我们创建了一个PyTorch张量tensor，然后使用numpy函数将其转换为Numpy数组。需要注意的是，由于PyTorch张量的数据类型是torch.int64，因此转换后的Numpy数组的数据类型也是int64。

print(tensor.dtype) # torch.int64
print(arr.dtype) # int64  

有坑

在进行Numpy数组和Torch张量的转换时，可能会遇到一些坑。
下面是两个实例，演示如何解决这些坑。

示例1：Numpy数组转换为PyTorch张量时数据类型不匹配

import numpy as np
import torch
# 创建一个Numpy数组
arr = np.array([1, 2, 3, 4, 5], dtype=np.float32)

# 创建一个PyTorch张量
tensor = torch.tensor(arr, dtype=torch.float32)

print(tensor)  # tensor([1., 2., 3., 4., 5.])

通过强制dtype指定格式，来避免数据类型不匹配的问题。

示例2：PyTorch张量转换为Numpy数组时需要使用detach函数

import numpy as np
import torch

# 创建一个PyTorch张量
tensor = torch.tensor([1, 2, 3, 4, 5])

# 将PyTorch张量转换为Numpy数组
arr = tensor.numpy()

print(arr) 

和前面转换是一样的代码，在有些情况下会报错：

‘Tensor’ object has no attribute ‘numpy’

在上面的示例中，我们创建了一个PyTorch张量tensor，然后使用numpy函数将其转换为Numpy数组。由于PyTorch张量是动态图，可能存在梯度计算操作，因此不能直接使用numpy函数进行转换。解决这个问题方法是，使用detach函数将张量从计算图中分离出来，然后再使用numpy函数进行转换。

import numpy as np
import torch

# 创建一个PyTorch张量
tensor = torch.tensor([1, 2, 3, 4, 5])

# 将PyTorch张量转换为Numpy数组
arr = tensor.detach().numpy()

print(arr)  # [1 2 3 4 5]

在上面的示例中，我们创建了一个PyTorch张量tensor，然后使用detach函数将其从计算图中分离出来，然后再使用numpy函数进行转换。