Python PyTorch Multiplexer用法及代码示例

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

用法:

class torchdata.datapipes.iter.Multiplexer(*datapipes)

参数：

datapipes-可迭代DataPipes将轮流产生它们的元素，直到它们全部耗尽

从每个输入 Iterable DataPipes (函数名称：mux )一次生成一个元素。例如，第一个输入 DataPipe 中的一个元素，然后是下一次迭代中第二个 DataPipe 中的一个元素，依此类推。它会跳过已耗尽的DataPipes，并在所有输入DataPipes耗尽时结束。

示例

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