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]