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]