[DataPipe] export fork, mux, demux for public usage (#64279)

Summary: Pull Request resolved: #64279 cc VitalyFedyunin ejguan Test Plan: Imported from OSS Reviewed By: ejguan Differential Revision: D30671971 Pulled By: NivekT fbshipit-source-id: 056ac12ef7183b254d1eec341145594639e47ef6
pytorch · Aug 31, 2021 · 6c8cb9b · 6c8cb9b
1 parent 491bf7c
commit 6c8cb9b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch/utils/data/datapipes/iter/__init__.py b/torch/utils/data/datapipes/iter/__init__.py
@@ -54,14 +54,17 @@
            'BucketBatcher',
            'Collator',
            'Concater',
+           'Demultiplexer',
            'FileLister',
            'FileLoader',
            'Filter',
+           'Forker',
            'Grouper',
            'HttpReader',
            'IterableWrapper',
            'LineReader',
            'Mapper',
+           'Multiplexer',
            'RoutedDecoder',
            'Sampler',
            'Shuffler',