ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


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.