tensor转换为numpy
时间: 2023-10-19 09:31:10 浏览: 88
tensor与numpy转换资源合集
5星 · 资源好评率100%
可以使用Tensor的`numpy()`方法将其转换为NumPy数组。例如:
```python
import torch
# 创建一个Tensor
x = torch.tensor([[1, 2], [3, 4]])
# 将Tensor转换为NumPy数组
x_np = x.numpy()
print(x_np)
```
输出为:
```
array([[1, 2],
[3, 4]], dtype=int64)
```
需要注意的是,如果Tensor在GPU上,则需要先将其移动到CPU上再进行转换。可以使用`to()`方法进行移动。例如:
```python
import torch
# 创建一个在GPU上的Tensor
x_gpu = torch.tensor([[1, 2], [3, 4]]).cuda()
# 将Tensor移动到CPU上
x_cpu = x_gpu.to('cpu')
# 将Tensor转换为NumPy数组
x_np = x_cpu.numpy()
print(x_np)
```
输出为:
```
array([[1, 2],
[3, 4]], dtype=int64)
```
阅读全文