Python PyTorch IndexAdder用法及代碼示例

本文簡要介紹python語言中 torchdata.datapipes.iter.IndexAdder 的用法。

用法:

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

參數：

• source_datapipe-可迭代DataPipe被索引，其行/批次必須是Dict類型

• index_name-存儲數據索引的鍵名

使用(函數名稱：add_index)向現有 Iterable DataPipe 添加索引。 DataPipe 中的行或批次必須具有類型 Dict ；否則，將拋出NotImplementedError。數據的索引設置為提供的 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}]