tf.contrib.layers.flatten(*args, **kwargs)

tf.contrib.layers.flatten(*args, **kwargs)

See the guide: Layers (contrib) > Higher level ops for building neural network layers

Flattens the input while maintaining the batch_size.

Assumes that the first dimension represents the batch.


  • inputs: a tensor of size [batch_size, ...].
  • outputs_collections: collection to add the outputs.
  • scope: Optional scope for name_scope.


a flattened tensor with shape [batch_size, k]. Raises: * ValueError: if inputs.dense_shape is wrong.

Defined in tensorflow/contrib/framework/python/ops/