使用torch.from_numpy将NumPy数组转为PyTorch张量

小蓝 • 2024-12-12 13:15 • 编程

一、介绍

PyTorch是近年来备受瞩目的深度学习框架，由于其灵活性和易用性，在学术界和工业界都得到了广泛的应用。而NumPy是Python中用于科学计算的基础包，主要用于数组处理。将NumPy数组转换为PyTorch张量非常常见，尤其是在进行图像处理和机器学习任务时，需要频繁地进行这个操作。这时候，使用PyTorch提供的函数torch.from_numpy可以快速地完成这个转换。下面我们就来具体探讨一下这个函数的用法和注意事项。

二、torch.from_numpy的用法

torch.from_numpy是PyTorch中用于将NumPy数组转换为张量的函数，语法非常简单：

import torch
import numpy as np
 
np_array = np.ones((3, 3))
tensor = torch.from_numpy(np_array)

该例子中，我们首先利用NumPy创建了一个3×3的全1矩阵np_array，然后通过torch.from_numpy函数将其转换成了PyTorch张量。转换后的结果tensor的类型是torch.DoubleTensor，数值与np_array完全一致。

需要注意的是，torch.from_numpy是不会复制数据的。这意味着，如果你的NumPy数组np_array发生了变化，那么由它转换而来的PyTorch张量tensor也会相应地发生变化。如果你希望得到一份数据的副本，可以使用tensor.clone()，这样就可以避免因为原始数据变化导致的问题。

三、数据类型的转换

NumPy和PyTorch的数据类型并不总是一一对应的，所以在将NumPy数组转换为PyTorch张量时，需要进行类型的转换。PyTorch支持的数据类型较多，包括浮点数、整数、布尔值等等。以下是两个数据类型的对应关系：

NumPy类型：np.float32，PyTorch类型：torch.FloatTensor
NumPy类型：np.int32，PyTorch类型：torch.LongTensor
NumPy类型：np.bool，PyTorch类型：torch.BoolTensor
NumPy类型：np.uint8，PyTorch类型：torch.ByteTensor
……

需要注意的是，在类型转换时可能会发生精度损失，所以要根据具体的情况选择合适的类型。

四、梯度追踪与非梯度追踪张量的转换

在PyTorch中，张量可以分为需要梯度追踪的张量和不需要梯度追踪的张量，它们分别是torch.Tensor类型和torch.autograd.Variable类型。我们可以通过torch.Tensor.detach()将梯度追踪张量转换为非梯度追踪张量。在将NumPy数组转换为张量时，有时候我们需要将其转换为不需要梯度追踪的张量，可以使用torch.tensor代替torch.from_numpy来实现这个功能。以下是一个例子：

import torch
import numpy as np
 
np_array = np.ones((3, 3))
tensor = torch.tensor(np_array)
non_grad_tensor = tensor.detach()

在该例子中，我们首先利用NumPy创建了一个3×3的全1矩阵np_array，然后通过torch.tensor函数将其转换成了PyTorch张量tensor。接着，我们用detach()方法将其转换为非梯度追踪张量non_grad_tensor。

五、结语

使用torch.from_numpy将NumPy数组转为PyTorch张量是一个非常常见的操作。本文介绍了torch.from_numpy的用法、数据类型的转换以及梯度追踪与非梯度追踪张量的转换等几个方面，希望这些内容对读者能有所帮助。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/246452.html