ValueError：“平面”输入的形状未完全定义（神经网络）

Question

我尝试对我的LSTM模型使用Flatten，但是它不起作用并出现此错误：

  

ValueError：“平面”输入的形状未完全定义（获取（无，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