tensor转化为array
时间: 2023-09-04 15:11:45 浏览: 100
Pytorch中Tensor与各种图像格式的相互转化详解
5星 · 资源好评率100%
可以使用 `numpy()` 方法将 PyTorch 的 tensor 转换为 numpy 数组,然后使用 `tolist()` 方法将其转换为 Python 列表,最后使用 `array` 函数将其转换为 ndarray 数组。示例代码如下:
```python
import torch
import numpy as np
# 创建一个 PyTorch tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将 tensor 转换为 numpy 数组
numpy_array = tensor.numpy()
# 将 numpy 数组转换为 Python 列表
python_list = numpy_array.tolist()
# 将 Python 列表转换为 ndarray 数组
ndarray_array = np.array(python_list)
# 打印结果
print(ndarray_array)
```
输出结果如下:
```
[[1 2]
[3 4]]
```
阅读全文