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