expose more io api (#1423)

stephenyan1231 · fmassa · commit 418755e3cc24 · 2019-10-31T15:52:30.000+01:00
diff --git a/torchvision/io/__init__.py b/torchvision/io/__init__.py
@@ -1,8 +1,16 @@
 from .video import write_video, read_video, read_video_timestamps
-from ._video_opt import _read_video_from_file, _read_video_timestamps_from_file, _HAS_VIDEO_OPT
+from ._video_opt import (
+    _read_video_from_file,
+    _read_video_timestamps_from_file,
+    _read_video_from_memory,
+    _read_video_timestamps_from_memory,
+    _HAS_VIDEO_OPT,
+)
 
 
 __all__ = [
     'write_video', 'read_video', 'read_video_timestamps',
-    '_read_video_from_file', '_read_video_timestamps_from_file', '_HAS_VIDEO_OPT',
+    '_read_video_from_file', '_read_video_timestamps_from_file',
+    '_read_video_from_memory', '_read_video_timestamps_from_memory',
+    '_HAS_VIDEO_OPT',
 ]
