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

tf_agents.utils.eager_utils.dataset_iterator

View source on GitHub

Constructs a Dataset iterator.

tf_agents.utils.eager_utils.dataset_iterator(
    dataset
)

The method used to construct the iterator 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:

Returns:

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