keras - Keras 输入层大小

Question

我有流动的代码

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,7,1) 但是当我运行代码时，模型将输入作为 6 行，每行有 7 个值并预测 6 个结果。我希望它将 6,7,1 作为 1 个输入并预测 6 个结果

Move.shape()

结果是 (6,6) 但我希望它是 (6)

标签： keras