Save the date! Google I/O returns May 18-20 Register now

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.