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

Flatten 层

[source]

Flatten class

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.json 文件中找到的 image_data_format 值(如果存在).默认为 "channels_last".

示例:

>>> x = keras.Input(shape=(10, 64))
>>> y = keras.layers.Flatten()(x)
>>> y.shape
(None, 640)