快捷方式

多路复用器

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]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获得针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源