tf_agents.utils.eager_utils.get_next
Stay organized with collections
Save and categorize content based on your preferences.
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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-04-26 UTC.
[null,null,["Last updated 2024-04-26 UTC."],[],[],null,["# tf_agents.utils.eager_utils.get_next\n\n\u003cbr /\u003e\n\n|---------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/utils/eager_utils.py#L650-L667) |\n\nReturns the next element in a `Dataset` iterator. \n\n tf_agents.utils.eager_utils.get_next(\n iterator\n )\n\nThe syntax used to retrieve the next item is conditioned on whether Graph mode\nis enabled. `dataset_iterator` and `get_next` are useful when we need to\nconstruct an iterator and iterate through it inside a `tensorflow.function`.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|------------|---------------------------------------------------------------------------------------------------------------------------------------------------|\n| `iterator` | a [`tf.data.Iterator`](https://www.tensorflow.org/api_docs/python/tf/data/Iterator) if in Graph mode; a `tf.data.EagerIterator` if in eager mode. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A [`tf.data.Iterator`](https://www.tensorflow.org/api_docs/python/tf/data/Iterator) if Graph mode is enabled; a Python iterator if in eager mode. ||\n\n\u003cbr /\u003e"]]