展平层

[源]

Flatten

keras.layers.Flatten(data_format=None, **kwargs)

将输入展平。不影响批量大小。

注意:如果输入的形状为 (batch,) 且没有特征轴,则展平操作会添加一个额外的通道维度,输出形状变为 (batch, 1)

参数

  • data_format:一个字符串,可以是 "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)