Спецификация ввода Keras для векторов word2vec

Question

Я прочитал все остальные ответы по этой теме, но мой вариант использования немного отличается.

У меня есть массив форм (800,128,1). Каждый элемент в 800 элементах хранит вложение word2vec формы (128,1). Теперь я хотел отправить это в качестве входных данных с размером пакета 64 в модель Keras с первым слоем в качестве входного слоя. Я получаю следующую ошибку:

expected party to have 2 dimensions, but got array with shape (800, 128, 1)

Я понимаю, что для входного слоя требуется 2 измерения, но какие два? Или я должен указать форму ввода самостоятельно как трехмерную?

В настоящее время входной слой:

Input(shape = (embedding_size, ), name = 'party')

Answer 1

Форма ввода - (embedding_size,), где embedding_size - предположительно 128. Таким образом, ожидается, что вход будет представлять собой массив формы (batch_size, embedding_size), а не (batch_size, embedding_size, 1). Вам нужно изменить форму массива, чтобы пропустить последнее измерение размера 1.