Cycler

class torchdata.datapipes.iter.Cycler(source_datapipe: IterDataPipe[T_co], count: Optional[int] = None)

默认情况下，无限循环指定的输入，或者循环指定的次数（函数名称：cycle）。

如果顺序无关紧要（例如，因为您计划稍后shuffle）或如果您希望在转到下一个元素之前多次重复一个元素，请使用Repeater。

参数：

• source_datapipe – 将循环遍历的源 DataPipe

• count – 读取source_datapipe` (如果 ``None，它将无限循环)的次数

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp = IterableWrapper(range(3))
>>> dp = dp.cycle(2)
>>> list(dp)
[0, 1, 2, 0, 1, 2]