tf.contrib.eager.Iterator

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.
Args
dataset A tf.data.Dataset object.
Raises
TypeError If dataset is an unsupported type.
RuntimeError When invoked without eager execution enabled.
Attributes
element_spec The type specification of an element of this iterator.
output_classes Returns the class of each component of an element of this iterator. (deprecated)

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

output_shapes Returns the shape of each component of an element of this iterator. (deprecated)
output_types Returns the type of each component of an element of this iterator. (deprecated)

Methods

get_next

View source

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

View source

Returns a nested structure of Tensors containing the next element.

__iter__

View source

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/eager/Iterator