Как использовать np.transpose()

Я пытаюсь понять, как правильно использовать np.transpose().

У меня есть массив np формы (4,28,28,8,8). Итак, это 4 изображения формы (224,224), которые я просмотрел, как показано предыдущей фигурой.

Я хотел бы вернуться к (4,224,224). Я считаю, что лучший способ - использовать функции np.transpose() и reshape(). Но я натыкаюсь на контрольно-пропускной пункт, как правильно вернуться назад.

Помощь. Пожалуйста и спасибо.

РЕДАКТИРОВАТЬ: (4 224 224) равно 4 (эта переменная может быть изменена, поскольку количество загружаемых изображений может измениться, может быть 4, может быть 1000) изображений формы (224 224). Я использовал listdir() для загрузки изображений. Во время загрузки я изменил размер до текущей формы (224 224). Я собираюсь выполнить операции над формой (4,28,28,8,8), которая технически представляет собой 4 изображения формы (224,224), разбитые на (28,28) блоки, каждый из которых содержит (8,8) блоков. Эту форму я получил, используя view_as_blocks, предоставленный scikit-image. Как только я выполню операции, требующие этой формы, я должен вернуться к (4 224 224). Где я застрял.