枚举器

class torchdata.datapipes.iter.Enumerator(source_datapipe: IterDataPipe[K], starting_index: int = 0)

通过枚举为现有数据管道添加索引，默认情况下索引从 0 开始（函数名称：enumerate）。

参数：:

• source_datapipe – 正在索引的可迭代数据管道

• starting_index – 枚举开始的索引

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp = IterableWrapper(['a', 'b', 'c'])
>>> enum_dp = dp.enumerate()
>>> list(enum_dp)
[(0, 'a'), (1, 'b'), (2, 'c')]