[feature request/1 min work] import Subset under namespace torch.utils.data #5922
Description
This is convenient when we need to manually split a validation set from a Dataset instance.
diff --git a/torch/utils/data/__init__.py b/torch/utils/data/__init__.py
index f3b008fb1..53da18468 100644
--- a/torch/utils/data/__init__.py
+++ b/torch/utils/data/__init__.py
@@ -1,3 +1,3 @@
-from .dataset import Dataset, TensorDataset, ConcatDataset
+from .dataset import Dataset, TensorDataset, ConcatDataset, Subset
from .dataloader import DataLoader