Python PyTorch Filter用法及代码示例

本文简要介绍python语言中 torchdata.datapipes.iter.Filter 的用法。

用法:

class torchdata.datapipes.iter.Filter(datapipe: IterDataPipe, filter_fn: Callable, drop_empty_batches: bool = True)

参数：

• datapipe-可迭代 DataPipe 被过滤

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

• drop_empty_batches-默认情况下，如果过滤后为空，则丢弃一个批次，而不是保留一个空列表

根据输入filter_fn(函数名称：filter)从源数据管道中过滤出元素。

示例

>>> 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]