切片器

class torchdata.datapipes.iter.Slicer(datapipe: IterDataPipe, index: Union[int, List[Hashable]], stop: Optional[int] = None, step: Optional[int] = None)

通过 start/stop/step 或索引返回输入 DataPipe 中元素的切片（函数名：slice）。

参数：:

• datapipe – 具有可迭代元素的 IterDataPipe

• index –

  切片的单个起始索引或要返回的索引列表（而不是 start/stop 切片）

  • 整数用于列表/元组。

  • 键用于字典。

• stop – 切片的结束位置。如果 index 是列表或元素是字典，则忽略。

• step – 从起始位置到结束位置要执行的步长。如果 index 是列表或元素是字典，则忽略。

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp = IterableWrapper([(0, 10, 100), (1, 11, 111), (2, 12, 122), (3, 13, 133), (4, 14, 144)])
>>> slice_dp = dp.slice(0, 2)
>>> list(slice_dp)
[(0, 10), (1, 11), (2, 12), (3, 13), (4, 14)]