Python tf.keras.layers.Flatten用法及代码示例

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

用法

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

data_format 一个字符串，是 channels_last (默认)或 channels_first 之一。输入中维度的排序。 channels_last 对应于形状为 (batch, ..., channels) 的输入，而 channels_first 对应于形状为 (batch, channels, ...) 的输入。它默认为您的 Keras 配置文件中的 image_data_format 值 ~/.keras/keras.json 。如果您从未设置它，那么它将是"channels_last"。

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

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)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.Flatten。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。