将张量转换为numpy数组
时间: 2023-11-23 18:07:24 浏览: 79
pytorch学习内容,包括Pytorch基本内容介绍,张量概念、张量的操作、张量与Numpy数组的相互转换、自动微分、神将网络
可以使用Tensor.numpy()方法将张量转换为numpy数组,例如:
```python
import torch
# 创建张量
x = torch.Tensor([[1, 2], [3, 4]])
# 将张量转换为numpy数组
x_numpy = x.numpy()
print(x_numpy)
```
输出结果为:
```
array([[1., 2.],
[3., 4.]], dtype=float32)
```
阅读全文