torch.Tensor怎么转换成numpy.ndarray
时间: 2024-02-11 16:07:41 浏览: 132
pytorch 实现tensor与numpy数组转换
可以使用 `numpy()` 函数将 `torch.Tensor` 转换为 `numpy.ndarray`,示例代码如下:
```python
import torch
import numpy as np
# 创建一个 tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将 tensor 转换为 numpy 数组
array = tensor.numpy()
print(array)
```
输出结果为:
```
array([[1, 2],
[3, 4]])
```
阅读全文