tensor如何转numpy
时间: 2023-10-19 19:13:08 浏览: 100
pytorch 实现tensor与numpy数组转换
您可以使用`numpy()`方法将Tensor转换为NumPy数组。以下是一个示例:
```python
import torch
# 创建一个Tensor
x = torch.tensor([1, 2, 3])
# 将Tensor转换为NumPy数组
numpy_array = x.numpy()
print(numpy_array)
```
输出:
```
[1 2 3]
```
请注意,将Tensor转换为NumPy数组时,如果Tensor在CPU上,则NumPy数组将与Tensor共享相同的内存。这意味着对NumPy数组的更改也会反映在原始的Tensor上。
阅读全文