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.
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].
ValueError: If inputs rank is unknown or less than 2.