# tf_agents.utils.eager_utils.dataset_iterator

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/utils/eager_utils.py#L627-L647) |

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`.

<br />

<br />

<br />

| ## Args ||
|---|---|
| `dataset` | a [`tf.data.Dataset`](https://www.tensorflow.org/api_docs/python/tf/data/Dataset). |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| 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. ||

<br />