Keras输入层大小

我有流动密码

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）