tf.contrib.layers.flatten

Flattens the input while maintaining the batch_size.

tf.contrib.layers.flatten(
    inputs, outputs_collections=None, scope=None
)

Assumes that the first dimension represents the batch.

Args
inputs	A tensor of size [batch_size, ...].
outputs_collections	Collection to add the outputs.
scope	Optional scope for name_scope.

Returns
A flattened tensor with shape [batch_size, k].

Raises
ValueError	If inputs rank is unknown or less than 2.