重复器

class torchdata.datapipes.iter.Repeater(source_datapipe: IterDataPipe[T_co], times: int)

在移至下一个元素之前，重复地生成源数据管道中的每个元素指定的次数（函数名：repeat）。请注意，此数据管道中没有进行任何复制操作，相同的元素会重复地生成。

如果您想多次按顺序生成整个数据管道，请使用 Cycler.

参数:

• source_datapipe – 将遍历的源数据管道

• times – 在移至下一个元素之前生成 source_datapipe 元素的次数

示例

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