当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.compat.v1.data.TFRecordDataset.make_initializable_iterator用法及代码示例


用法

make_initializable_iterator(
    shared_name=None
)

参数

  • shared_name (可选。)如果非空,则返回的迭代器将以给定名称在共享相同设备的多个会话中共享(例如,当使用远程服务器时)。

返回

抛出

  • RuntimeError 如果启用了即刻执行。

为此数据集的元素创建迭代器。 (已弃用)

警告:此函数已弃用。它将在未来的版本中删除。更新说明:这是一个已弃用的 API,仅应在 TF 1 图形模式和通过 tf.compat.v1 可用的旧版 TF 2 图形模式中使用。在所有其他情况下 - 即渴望模式和内部 tf.function - 您可以使用 for elem in dataset:... 或通过显式创建迭代器 iterator = iter(dataset) 并通过 values = next(iterator) 获取其元素来使用数据集元素。此外,此 API 在 TF 2 中不可用。在从 TF 1 过渡到 TF 2 期间,您可以使用 tf.compat.v1.data.make_initializable_iterator(dataset) 为通过 TF 2 API 创建的数据集创建 TF 1 图形模式样式迭代器。请注意,这应该是您的代码库的暂时状态,因为通常不能保证 TF 1 和 TF 2 代码的互操作性。

注意:返回的迭代器将处于未初始化状态,您必须在使用前运行iterator.initializer 操作:

# Building graph ...
dataset = ...
iterator = dataset.make_initializable_iterator()
next_value = iterator.get_next()  # This is a Tensor.

# ... from within a session ...
sess.run(iterator.initializer)
try:
  while True:
    value = sess.run(next_value)
    ...
except tf.errors.OutOfRangeError:
    pass

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.data.TFRecordDataset.make_initializable_iterator。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。