keras.layers.Flatten讲解
程序员文章站
2024-03-15 11:22:41
...
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])
上一篇: Docker 之间如何共享数据?