как использовать np.transpose () для формы (1, 7, 7, 1024)

Question

a = np.ones((1, 7, 7, 1024))

Я хочу изменить форму на (1, 1024, 7, 7)

Я видел учебник на сайте Numpy, я подумал, что нужно установить позициюсоответственно. Поэтому я использовал код ниже.

shape_ = np.transpose(a, (0, 2, 3, 1))

Когда я распечатаю форму:

Форма будет: (1, 7, 1024, 7)

Я могуНе могу понять причины этого.

Спасибо за помощь.

Answer 1

Правильный ответ:

shape_ = np.transpose(a, (0, 3, 2, 1))

При печати вывода выдается следующее:

(1, 1024, 7, 7)

0 - первая ось, 1 - вторая, 2 - третьяи так далее в зависимости от количества осей, так что вы просто переставляете их по своему усмотрению.