Преобразование 3D-массива в 2D

#deep-learning

#глубокое обучение

Вопрос:

У меня есть (5, 128, 768) матрица, то есть у меня есть 5 вложенных пространств формы (128, 768) . Поскольку все они сохраняют связь, и ради моей модели мне нужно объединить их в уникальный результат: (1, 128, 768*5) . Если я просто объединю их все вместе axis=-1 , я потеряю какую-то информацию?

Создание такого объединения-единственный способ, которым я могу придумать решение этой проблемы. Есть ли какой-нибудь лучший вариант?