ValueError: форма ввода для «Flatten» не полностью определена (нейронные сети)

Question

Я пытаюсь использовать Flatten для моей модели LSTM, но это не работает, и я получаю эту ошибку:

ValueError: Форма ввода для "Flatten" не полностью определена (получено(Нет, 450) Обязательно передайте полный аргумент "input_shape" или "batch_input_shape" первому слою в вашей модели.

Модель:

input_vector_size = kwargs.get('input_vector_size', 128)
dense_size = kwargs.get('dense_size', 20)
timesteps = None
xav_init = tf.contrib.layers.xavier_initializer()
##########

model = Sequential()
model.add(LSTM(int(input_vector_size), input_shape=timesteps,int(input_vector_size)), return_sequences=True))
model.add(SeqSelfAttention(attention_activation='softmax'))
model.add(Flatten())
model.add(Dense(dense_size, activation='sigmoid', kernel_initializer=xav_init))
model.add(Dense(kwargs.get('label_size', 2), activation='sigmoid'))

Как мнерешить эту проблему?

Answer 1

Просто для тех, кто хочет знать ответ.Вы должны изменить input_shape на (1, int (input_vector_size)), чтобы использовать Flatten