tf.layers.flatten

tf.layers.flatten(
    inputs,
    name=None
)

Defined in tensorflow/python/layers/core.py.

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

Arguments:

  • inputs: Tensor input.
  • name: The name of the layer (string).

Returns:

Reshaped tensor.

Examples:

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

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