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


View source on GitHub

Get static input batch size if available, with fallback to the dynamic one.

flat_input An iterable of time major input Tensors of shape [max_time, batch_size, ...]. All inputs should have compatible batch sizes.

The batch size in Python integer if available, or a scalar Tensor otherwise.

ValueError if there is any input with an invalid shape.