Mapper

class torchdata.datapipes.map.Mapper(datapipe: ~MapDataPipe, fn: ~Callable = <function default_fn>)

对来自源 DataPipe 的每个项目应用输入函数（函数名称：map）。

该函数可以是任何常规 Python 函数或部分对象。不建议使用 lambda 函数，因为它不受 pickle 支持。

参数：

• datapipe – 源 MapDataPipe

• fn – 应用于每个项目的函数

示例

>>> # xdoctest: +SKIP
>>> from torchdata.datapipes.map import SequenceWrapper, Mapper
>>> def add_one(x):
...     return x + 1
>>> dp = SequenceWrapper(range(10))
>>> map_dp_1 = dp.map(add_one)
>>> list(map_dp_1)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> map_dp_2 = Mapper(dp, lambda x: x + 1)
>>> list(map_dp_2)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]