过滤器

class torchdata.datapipes.iter.Filter(datapipe: IterDataPipe[T_co], filter_fn: Callable, input_col=None)

根据输入 filter_fn（函数名称：filter）过滤源 datapipe 中的元素。

参数：

• datapipe – 正在过滤的可迭代 DataPipe

• filter_fn – 自定义函数，将元素映射到布尔值。

• input_col –

  应用 filter_fn 的数据索引或索引，例如

  • None 作为默认值，将 filter_fn 直接应用于数据。

  • 整数用于列表/元组。

  • 键用于字典。

示例

>>> # xdoctest: +SKIP
>>> from torchdata.datapipes.iter import IterableWrapper
>>> def is_even(n):
...     return n % 2 == 0
>>> dp = IterableWrapper(range(5))
>>> filter_dp = dp.filter(filter_fn=is_even)
>>> list(filter_dp)
[0, 2, 4]