tensor转numpy
时间: 2023-10-19 14:11:51 浏览: 40
tensor与numpy转换资源合集
5星 · 资源好评率100%
请使用以下代码将 Tensor 转换为 NumPy 数组:
```python
import numpy as np
import torch
# 创建一个 Tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将 Tensor 转换为 NumPy 数组
array = tensor.numpy()
print(array) # 输出 [[1 2], [3 4]]
```
阅读全文