tf.contrib.eager.Iterator
An iterator producing tf.Tensor objects from a tf.data.Dataset.
tf.contrib.eager.Iterator( 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 |
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
get_next( name=None )
Returns a nested structure of tf.Tensor
s 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 Tensor
s containing the next element.
__iter__
__iter__()
© 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