tf.layers.flatten

View source on GitHub

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

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.layers.flatten

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

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).

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)`