Shortcuts

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

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources