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.