• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Functional DataPipe
2from torch.utils.data.datapipes.map.callable import MapperMapDataPipe as Mapper
3from torch.utils.data.datapipes.map.combinatorics import (
4    ShufflerIterDataPipe as Shuffler,
5)
6from torch.utils.data.datapipes.map.combining import (
7    ConcaterMapDataPipe as Concater,
8    ZipperMapDataPipe as Zipper,
9)
10from torch.utils.data.datapipes.map.grouping import BatcherMapDataPipe as Batcher
11from torch.utils.data.datapipes.map.utils import (
12    SequenceWrapperMapDataPipe as SequenceWrapper,
13)
14
15
16__all__ = ["Batcher", "Concater", "Mapper", "SequenceWrapper", "Shuffler", "Zipper"]
17
18# Please keep this list sorted
19assert __all__ == sorted(__all__)
20