tf.contrib.layers.flatten

View source on GitHub

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

ValueError If inputs rank is unknown or less than 2.