TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.keras.layers.Flatten

TensorFlow 1 version View source on GitHub

Class Flatten

Flattens the input. Does not affect the batch size.

Inherits From: Layer

Aliases:

  • Class tf.compat.v1.keras.layers.Flatten
  • Class tf.compat.v2.keras.layers.Flatten

Used in the guide:

Used in the tutorials:

If inputs are shaped (batch,) without a channel dimension, then flattening adds an extra channel dimension and output shapes are (batch, 1).

Arguments:

  • data_format: A string, one of channels_last (default) or channels_first. The ordering of the dimensions in the inputs. channels_last corresponds to inputs with shape (batch, ..., channels) while channels_first corresponds to inputs with shape (batch, channels, ...). It defaults to the image_data_format value found in your Keras config file at ~/.keras/keras.json. If you never set it, then it will be "channels_last".

Example:

model = Sequential()
model.add(Convolution2D(64, 3, 3,
                        border_mode='same',
                        input_shape=(3, 32, 32)))
# now: model.output_shape == (None, 64, 32, 32)

model.add(Flatten())
# now: model.output_shape == (None, 65536)

__init__

View source

__init__(
    data_format=None,
    **kwargs
)