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

NivekT · facebook-github-bot · commit 6c8cb9bd7629 · 2021-08-31T14:34:30.000-07:00
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
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',
