Dropper

class torchdata.datapipes.iter.Dropper(datapipe: IterDataPipe, indices: Union[Hashable, List[Hashable]])

通过其索引删除输入 DataPipe 中的列/元素（函数名称：drop）。

参数：

• datapipe – 带有要删除的列的 IterDataPipe

• indices –

  要删除的单个列索引或索引列表

  • 整数用于列表/元组。

  • 键用于字典。

示例

>>> from torchdata.datapipes.iter import IterableWrapper, ZipperMapDataPipe
>>> dp1 = IterableWrapper(range(5))
>>> dp2 = IterableWrapper(range(10, 15))
>>> dp = dp1.zip(dp2)
>>> list(dp)
[(0, 10), (1, 11), (2, 12), (3, 13), (4, 14)]
>>> drop_dp = dp.drop(1)
>>> list(drop_dp)
[(0), (1), (2), (3), (4)]