Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.eager_utils.get_next

View source on GitHub

Returns the next element in a Dataset iterator.

tf_agents.utils.eager_utils.get_next(
    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.

Args:

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

Returns:

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