Keras 输入层大小

Question

我有流动的代码

java.lang.Object

输入形状为：(6,7,1) 但是当我运行代码时，模型将输入作为 6 行，每行有 7 个值并预测 6 个结果。 我希望它将 6,7,1 作为 1 个输入并预测 6 个结果

AI = Sequential()
AI.add(Dense(128, input_shape=(6,7,1)))
AI.add(Flatten())
AI.add(Dense(6, activation='sigmoid'))
AI.compile(Adam(lr=0.5),'binary_crossentropy')
AI.summary()
Move = AI.predict([Input])

导致 (6,6) 但我希望它是 (6)