Multiplexer

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

Yields one element at a time from each of the input Iterable DataPipes (functional name: mux). As in, one element from the 1st input DataPipe, then one element from the 2nd DataPipe in the next iteration, and so on. It ends when the shortest input DataPipe is exhausted.

Parameters:

datapipes – Iterable DataPipes that will take turn to yield their elements, until the shortest DataPipe is exhausted

Example

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