Cycler¶

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

Cycles the specified input in perpetuity by default, or for the specified number of times (functional name: cycle).

Parameters:

source_datapipe – source DataPipe that will be cycled through
count – the number of times to read through source_datapipe` (if ``None, it will cycle in perpetuity)

Example

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

Docs