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


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


用法

enumerate(
    start=0, name=None
)

参数

  • start 一个 tf.int64 标量 tf.Tensor ,表示枚举的起始值。
  • name 可选的。 enumerate 使用的 tf.data 操作的名称。

返回

  • Dataset 一个Dataset

枚举此数据集的元素。

它类似于 python 的 enumerate

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
dataset = dataset.enumerate(start=5)
for element in dataset.as_numpy_iterator():
  print(element)
(5, 1)
(6, 2)
(7, 3)
# The (nested) structure of the input dataset determines the
# structure of elements in the resulting dataset.
dataset = tf.data.Dataset.from_tensor_slices([(7, 8), (9, 10)])
dataset = dataset.enumerate()
for element in dataset.as_numpy_iterator():
  print(element)
(0, array([7, 8], dtype=int32))
(1, array([ 9, 10], dtype=int32))

相关用法


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