展平N维numpy数组以形成K维数组（K≠1，K <n）

Question

=“”

假设我有一个三维数组，我将其称为A，其中第一维等于N。如何找到B，它等效于：

np.concatenate((A[0],A[1],A[2],...,A[N-1]),axis=1)

作为一个例子，假设我有：

A = np.array([[[1,2,3,4],[5,6,7,8]],[[9,10,11,12],[13,14,15,16]],[[17,18,19,20],[21,22,23,24]]])

print(A)
[[[ 1  2  3  4]
  [ 5  6  7  8]]

 [[ 9 10 11 12]
  [13 14 15 16]]

 [[17 18 19 20]
  [21 22 23 24]]]

如何（一行）返回B，它等于：

print(B)
[[ 1  2  3  4  9 10 11 12 17 18 19 20]
 [ 5  6  7  8 13 14 15 16 21 22 23 24]]

我尝试过重塑，但这不会返回所需的元素顺序。谢谢。