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