Использование может использовать функцию изменения формы для того же ... Для простоты я взял размеры меньше, но, конечно, это будет служить цели
arr = np.array([[[1,2,3],[4,5,6],[7,8,9]],[[9,8,7],[6,5,4],[3,2,1]]])
arr.shape
>> (2,3,3)
arr = np.reshape(arr,(3,3,2))
arr.shape
>> (3,3,2)
Надеюсь, это поможет вам.