keras.layers.Flatten(input_shape=[])
用于将输入层的数据压成一维的数据,一般用再卷积层和全连接层之间(因为全连接层只能接收一维数据,而卷积层可以处理二维数据,就是全连接层处理的是向量,而卷积层处理的是矩阵),原理如下
input=keras.layers.Input(shape=[28,28])
layers_Flatten=keras.layers.Flatten(input_shape=[28,28])(input)
layers1=keras.layers.Dense(100,activation='relu')(layers_Flatten)
layers2=keras.layers.Dense(200,activation='relu')(layers1)
layers3=keras.layers.Dense(100,activation='relu')(layers2)
output=keras.layers.Dense(10,activation='softmax')(layers3)
model=keras.models.Model(inputs=[input],
outputs=[output])