tf.contrib.eager.Iterator

Class Iterator

Defined in tensorflow/contrib/eager/python/datasets.py.

An iterator producing tf.Tensor objects from a tf.data.Dataset.

NOTE: Unlike the iterator created by the tf.data.Dataset.make_one_shot_iterator method, this class enables additional experimental functionality, such as prefetching to the GPU.

__init__

__init__(dataset)

Creates a new iterator over the given dataset.

For example:

dataset = tf.data.Dataset.range(4)
for x in Iterator(dataset):
  print(x)

Tensors produced will be placed on the device on which this iterator object was created.

Args:

• dataset: A tf.data.Dataset object.

Raises:

• TypeError: If dataset is an unsupported type.
• RuntimeError: When invoked without eager execution enabled.

Properties

output_classes

Returns the class of each component of an element of this iterator.

The expected values are tf.Tensor and tf.SparseTensor.

Returns:

A nested structure of Python type objects corresponding to each component of an element of this dataset.

output_shapes

Returns the shape of each component of an element of this iterator.

Returns:

A nested structure of tf.TensorShape objects corresponding to each component of an element of this dataset.

output_types

Returns the type of each component of an element of this iterator.

Returns:

A nested structure of tf.DType objects corresponding to each component of an element of this dataset.

Methods

tf.contrib.eager.Iterator.__iter__

__iter__()

tf.contrib.eager.Iterator.__next__

__next__()

tf.contrib.eager.Iterator.get_next

get_next(name=None)

Returns a nested structure of tf.Tensors containing the next element.

Args:

• name: (Optional.) A name for the created operation. Currently unused.

Returns:

A nested structure of tf.Tensor objects.

Raises:

tf.errors.OutOfRangeError: If the end of the dataset has been reached.

tf.contrib.eager.Iterator.next

next()

Returns a nested structure of tf.Tensors containing the next element.