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.
|
View source on GitHub