Python PyTorch FlatMapper用法及代碼示例

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

用法:

class torchdata.datapipes.iter.FlatMapper(datapipe: IterDataPipe, fn: Callable)

參數：

• datapipe-來源IterDataPipe

• fn-要應用於 DataPipe 中每個元素的函數，輸出必須是序列

對源 DataPipe 中的每個項目應用函數，然後將輸出展平為單個未嵌套的 IterDataPipe (函數名稱： flatmap )。

注意

fn 的輸出必須是一個序列。否則，將引發錯誤。

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> def fn(e):
>>>     return [e, e * 10]
>>> source_dp = IterableWrapper(list(range(5)))
>>> flatmapped_dp = source_dp.flatmap(fn)
>>> list(flatmapped_dp)
[0, 0, 1, 10, 2, 20, 3, 30, 4, 40]