Python PyTorch Zipper用法及代码示例

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

用法:

class torchdata.datapipes.iter.Zipper(*datapipes: IterDataPipe)

参数：

*datapipes-可迭代DataPipes正在聚合

将每个输入 DataPipes 中的元素聚合到一个元组中(函数名称：zip )。一旦最短输入DataPipe耗尽，输出就会停止。

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp1, dp2, dp3 = IterableWrapper(range(5)), IterableWrapper(range(10, 15)), IterableWrapper(range(20, 25))
>>> list(dp1.zip(dp2, dp3))
[(0, 10, 20), (1, 11, 21), (2, 12, 22), (3, 13, 23), (4, 14, 24)]