tf.keras.layers.Flatten
使输入展平,不会影响批处理的大小。如果输入形状为(batch,),没有特征轴,则展平会增加额外的通道尺寸,输出形状为(batch,1)。
tf.keras.layers.Flatten(
data_format=None, **kwargs
)
data_format:一个字符串,是channels _ last(默认)还是channels _ first。输入中尺寸的顺序。channels_last对应于具有形状的输入(batch, ..., channels)而channels _ first对应于具有形状的输入(batch, channels, ...).
例如: