torch.utils.data.dataloader — PyTorch 1.10.1 documentation
pytorch.org › torch › utilsclass DataLoader (Generic [T_co]): r """ Data loader. Combines a dataset and a sampler, and provides an iterable over the given dataset. The :class:`~torch.utils.data.DataLoader` supports both map-style and iterable-style datasets with single- or multi-process loading, customizing loading order and optional automatic batching (collation) and memory pinning.