Expose documentation for random_split (#7676)

zou3519 · soumith · commit e37da05bd5b0 · 2018-05-18T17:16:25.000+02:00
Fixes #7640
diff --git a/docs/source/data.rst b/docs/source/data.rst
@@ -12,3 +12,4 @@ torch.utils.data
 .. autoclass:: torch.utils.data.SubsetRandomSampler
 .. autoclass:: torch.utils.data.WeightedRandomSampler
 .. autoclass:: torch.utils.data.distributed.DistributedSampler
+.. autofunction:: torch.utils.data.dataset.random_split
