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

tf.contrib.rnn.best_effort_input_batch_size

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.