Python PyTorch Cycler用法及代碼示例

本文簡要介紹python語言中 torchdata.datapipes.iter.Cycler 的用法。

用法:

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

參數：

• source_datapipe-將循環通過的源DataPipe

• count-讀取 source_datapipe` (if ``None 的次數，它將永久循環)

默認情況下永久循環指定的輸入，或循環指定的次數(函數名稱：cycle)。

示例

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