展平层

[源代码]

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)