Flatten
类keras.layers.Flatten(data_format=None, **kwargs)
将输入展平。不影响批量大小。
注意:如果输入的形状为 (batch,)
且没有特征轴,则展平操作会添加一个额外的通道维度,输出形状变为 (batch, 1)
。
参数
"channels_last"
(默认) 或 "channels_first"
。输入中维度的顺序。"channels_last"
对应于形状为 (batch, ..., channels)
的输入,而 "channels_first"
对应于形状为 (batch, channels, ...)
的输入。未指定时,使用 Keras 配置文件 ~/.keras/keras.json
(如果存在)中的 image_data_format
值。默认为 "channels_last"
。示例
>>> x = keras.Input(shape=(10, 64))
>>> y = keras.layers.Flatten()(x)
>>> y.shape
(None, 640)