把[7,7,128]的三维向量转化为[1,128]向量
时间: 2024-01-08 20:02:05 浏览: 51
python读取图像矩阵文件并转换为向量实例
可以使用numpy库中的reshape函数来实现:
```python
import numpy as np
vector_3d = np.array([7, 7, 128])
vector_2d = vector_3d.reshape(1, -1)
```
其中,reshape函数的第一个参数为新数组的形状,这里设为(1, -1),表示将原数组转化为一行,列数自动计算得出。因此,vector_2d为一个行向量,形状为(1, 128)。
阅读全文