numpy转成torch
时间: 2023-10-05 22:14:48 浏览: 93
Pyorch之numpy与torch之间相互转换方式
要将NumPy数组转换为PyTorch张量,可以使用`torch.from_numpy()`函数。以下是一个示例:
```python
import numpy as np
import torch
# 创建一个NumPy数组
numpy_array = np.array([1, 2, 3])
# 将NumPy数组转换为PyTorch张量
torch_tensor = torch.from_numpy(numpy_array)
print(torch_tensor)
```
输出:
```
tensor([1, 2, 3])
```
使用`torch.from_numpy()`函数可以将NumPy数组转换为PyTorch张量,并且它们共享相同的内存。这意味着对张量的修改也会反映在原始NumPy数组中。如果需要避免共享内存,可以使用`.clone()`方法创建张量的副本。
请注意,将大型NumPy数组转换为张量可能会占用大量内存,因此请确保计算机具有足够的内存来处理。
阅读全文