tf.compat.v1.layers.flatten

View source on GitHub

Flattens an input tensor while preserving the batch axis (axis 0). (deprecated)

tf.compat.v1.layers.flatten(
    inputs, name=None, data_format='channels_last'
)

Warning: THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Use keras.layers.Flatten instead.

Arguments:

• inputs: Tensor input.
• name: The name of the layer (string).
• 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, height, width, channels) while channels_first corresponds to inputs with shape (batch, channels, height, width).

Returns:

Reshaped tensor.

Examples:

x = tf.compat.v1.placeholder(shape=(None, 4, 4), dtype='float32')
  y = flatten(x)
  # now `y` has shape `(None, 16)`

  x = tf.compat.v1.placeholder(shape=(None, 3, None), dtype='float32')
  y = flatten(x)
  # now `y` has shape `(None, None)`