怎么将tensor转数组
时间: 2024-01-12 08:04:37 浏览: 62
将pytorch转成longtensor的简单方法
你可以使用 PyTorch 中的 `numpy()` 方法将 tensor 转换为 numpy 数组。示例如下:
```python
import torch
# 创建一个 2x2 的 tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将 tensor 转换为 numpy 数组
array = tensor.numpy()
print(array)
```
输出结果为:
```
array([[1, 2],
[3, 4]], dtype=int64)
```
注意,如果你想在 GPU 上创建 tensor,需要先将它转移到 CPU 上才能将其转换为 numpy 数组。可以使用 `cpu()` 方法将 tensor 转移到 CPU 上,如下所示:
```python
import torch
# 在 GPU 上创建一个 2x2 的 tensor
tensor = torch.tensor([[1, 2], [3, 4]]).cuda()
# 将 tensor 转移到 CPU 上,并将其转换为 numpy 数组
array = tensor.cpu().numpy()
print(array)
```
如果你正在使用其他的深度学习框架,可以查看相应的文档了解如何将 tensor 转换为数组。
阅读全文