IndexAdder

class torchdata.datapipes.iter.IndexAdder(source_datapipe: IterDataPipe[Dict], index_name: str = 'index')

向现有的可迭代 DataPipe 添加索引（函数名称：add_index）。DataPipe 中的行或批次必须是 Dict 类型；否则，将抛出 NotImplementedError。数据的索引设置为提供的 index_name。

参数：

• source_datapipe – 要索引的可迭代 DataPipe，其行/批次必须为 Dict 类型

• index_name – 用于存储数据索引的键的名称

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp = IterableWrapper([{'a': 1, 'b': 2}, {'c': 3, 'a': 1}])
>>> index_dp = dp.add_index("order")
>>> list(index_dp)
[{'a': 1, 'b': 2, 'order': 0}, {'c': 3, 'a': 1, 'order': 1}]