Как изменить и изменить ранг массива numpy?

Question

если у меня есть массив numpy формы

(16, 224, 224, 6)

как мне изменить его на

(224, 224, 6*16)

так что все элементы все еще в новой форме? PS: Мне нужен numpy ответ, а не ответ, полученный из pytorch, пожалуйста:)

Answer 1

Я думаю, вы можете использовать transpose и reshape:

a.transpose([1,2,3,0]).reshape(244,244,-1)