![]() |
Get static input batch size if available, with fallback to the dynamic one.
tf.contrib.rnn.best_effort_input_batch_size(flat_input)
Args:
flat_input
: An iterable of time major input Tensors of shape[max_time, batch_size, ...]
. All inputs should have compatible batch sizes.
Returns:
The batch size in Python integer if available, or a scalar Tensor otherwise.
Raises:
ValueError
: if there is any input with an invalid shape.