Tensorflow 2 - 如何从结构创建迭代器

【问题标题】：Tensorflow 2 - How to create iterator from structureTensorflow 2 - 如何从结构创建迭代器
【发布时间】：2020-11-29 18:27:06
【问题描述】：

以下行在 Tensorflow 2 中的等价物是什么：

tf.compat.v1.data.Iterator.from_structure(output_types, output_shapes)

换句话说，如何在不使用 v1 兼容模块的情况下从结构中创建迭代器？

【问题讨论】：

标签： python tensorflow iterator tensorflow2.0 tensorflow-datasets

【解决方案1】：

我的尝试并没有完全回答这个问题，它基于documentation。它将显示那里提到的不推荐使用的警告。

import tensorflow as tf
from tensorflow.python.data import Iterator

iterator = Iterator.from_structure(tf.int64, tf.TensorShape([]))
dataset_range = tf.data.Dataset.range(10)
range_initializer = iterator.make_initializer(dataset_range)
iterator = iter(dataset_range)
print(iterator.get_next())

iterator = iter(dataset_range)
print(iterator.get_next())

【讨论】：