tf_agents.utils.eager_utils.dataset_iterator
Stay organized with collections
Save and categorize content based on your preferences.
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.
Returns |
|---|
A tf.data.Iterator if Graph mode is enabled; a tf.data.EagerIterator 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.dataset_iterator\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#L627-L647) |\n\nConstructs a `Dataset` iterator. \n\n tf_agents.utils.eager_utils.dataset_iterator(\n dataset\n )\n\nThe method used to construct the iterator 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| `dataset` | a [`tf.data.Dataset`](https://www.tensorflow.org/api_docs/python/tf/data/Dataset). |\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 tf.data.EagerIterator if in eager mode. ||\n\n\u003cbr /\u003e"]]
View source on GitHub