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]