张量转换为numpy数组
时间: 2024-01-28 15:42:41 浏览: 77
pytorch学习内容,包括Pytorch基本内容介绍,张量概念、张量的操作、张量与Numpy数组的相互转换、自动微分、神将网络
可以使用numpy方法将张量转换为numpy数组。具体操作如下:
```python
import numpy as np
import torch
tensor = torch.randn(3, 4)
numpy_array = tensor.numpy()
```
其中,torch.randn(3, 4)创建一个形状为(3, 4)的随机张量,tensor.numpy()将其转换为numpy数组并返回。
阅读全文