Keras 2 API 文档 / 层 API / 重塑层 / Flatten 层

Flatten 层

[源]

Flatten

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

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

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

参数

  • data_format: 字符串类型,取值为 channels_last(默认)或 channels_first。输入的维度顺序。 channels_last 对应于形状为 (batch, ..., channels) 的输入,而 channels_first 对应于形状为 (batch, channels, ...) 的输入。未指定时,使用 TF-Keras 配置文件 ~/.keras/keras.json 中找到的 image_data_format 值(如果存在),否则为 'channels_last'。默认为 'channels_last'。

示例

>>> model = tf.keras.Sequential()
>>> model.add(tf.keras.layers.Conv2D(64, 3, 3, input_shape=(3, 32, 32)))
>>> model.output_shape
(None, 1, 10, 64)
>>> model.add(Flatten())
>>> model.output_shape
(None, 640)