| |
| 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'] |
|
|
| |
| assert __all__ == sorted(__all__) |
|
|