Watch talks from the 2019 TensorFlow Dev Summit Watch now

tf.contrib.rnn.best_effort_input_batch_size

tf.contrib.rnn.best_effort_input_batch_size(flat_input)

Defined in tensorflow/python/ops/rnn.py.

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

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.