numpy.reshape()

numpy で3次元配列を２次元配列にした場合の動作確認。

arr=np.array([[[1,2],[3,4],[5,6]],
              [[10,20],[30,40],[50,60]],
              [[100,200],[300,400],[500,600]],
              [[1000,2000],[3000,4000],[5000,6000]]])
              
print(arr.shape)
(4, 3, 2)

print(arr.reshape([4,-1]))
[[   1    2    3    4    5    6]
 [  10   20   30   40   50   60]
 [ 100  200  300  400  500  600]
 [1000 2000 3000 4000 5000 6000]]

転置すると、i,j,k が k,j,i になります。簡単。