tf_agents.utils.eager_utils.get_next

View source on GitHub

Returns the next element in a Dataset iterator.

The syntax used to retrieve the next item is conditioned on whether Graph mode is enabled. dataset_iterator and get_next are useful when we need to construct an iterator and iterate through it inside a tensorflow.function.

iterator a tf.data.Iterator if in Graph mode; a tf.data.EagerIterator if in eager mode.

A tf.data.Iterator if Graph mode is enabled; a Python iterator if in eager mode.