Как преобразовать размерность части массива?

Question

A = np.ones((14, 32, 3, 3))

Пусть у нас имеется 4-мерный NumPy массив.

Можно ли применить функцию np.ravel, или аналогичную ей, вдоль последних трех осей массива А?

Tо есть, в итоге должна получиться матрица с размерами (14, 32 * 3 * 3)).

Answer 1

можно воспользоваться методом ndarray.reshape():

A = np.ones((14, 32, 3, 3))
res = A.reshape(-1, np.prod(A.shape[1:]))

результат:

res.shape
>>> (14, 288)