多路复用器¶
- class torchdata.datapipes.iter.Multiplexer(*datapipes)¶
一次从每个输入可迭代数据管道(函数名称:
mux
)中生成一个元素。也就是说,在下次迭代中,先从第一个输入数据管道生成一个元素,然后从第二个输入数据管道生成一个元素,以此类推。当最短的输入数据管道耗尽时结束。
- 参数:
datapipes – 可迭代数据管道,将轮流生成其元素,直到最短的数据管道耗尽
示例
>>> # xdoctest: +REQUIRES(module:torchdata) >>> from torchdata.datapipes.iter import IterableWrapper >>> dp1, dp2, dp3 = IterableWrapper(range(3)), IterableWrapper(range(10, 15)), IterableWrapper(range(20, 25)) >>> list(dp1.mux(dp2, dp3)) [0, 10, 20, 1, 11, 21, 2, 12, 22]