Shortcuts

Slicer

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

returns a slice of elements in input DataPipe via start/stop/step or indices (functional name: slice).

Parameters:
  • datapipe – IterDataPipe with iterable elements

  • index – a single start index for the slice or a list of indices to be returned instead of a start/stop slice

  • stop – the slice stop. ignored if index is a list

  • step – step to be taken from start to stop. ignored if index is a list

Example

>>> 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)]

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