将三维numpy阵列合并到二维阵列

我有一个以下维度的numpy数组：

(2, 4, 20)

我想将其合并，以便最终得到：

(2, 80)

---

我试过

np.concatenate

，

np.dstack

，

np.vstack

和

np.hstack

，但都没有成功。我应该怎么做呢？

使用

np。重塑功能：

import numpy as np
A=np.random.rand(2,4,20) #creates a random narray for the example
A.shape
> (2,4,20) 
B = A.reshape(A.shape[0],A.shape[1]*A.shape[2])
B.shape
> (2,80)

重塑有不同的顺序选项，请参见。
也许可以看看
np。重塑