tf.contrib.eager.Iterator

Class Iterator

Inherits From: CheckpointableBase

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.

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

__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:

Raises:

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

__iter__

__iter__()

__next__

__next__()

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.

next

next()

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