mindspore.dataset.dataloader
This module provides iterators for loading datasets. It supports loading both Map Style and Iterable Style datasets, and offers multi-process concurrent loading.
Data Loader
Data loader provides an iterator over the given dataset. |
Datasets
Base class for map style datasets. |
|
Base class for iterable datasets. |
|
Each sample is retrieved by indexing the input tensors along their first dimension. |
Samplers
Base Class of the Sampler |
|
Samples the dataset elements sequentially. |
|
Samples the dataset elements randomly. |
|
A sampler that generates mini-batch indices each time. |
|
A sampler that partitioning datasets for distributed training. |
Collate Functions
Take in a batch of data and put the elements within the batch into a tensor with an additional outer dimension - batch size. |
|
Convert each NumPy array element into a |
|
General collate function that handles collection type of element within each batch. |
Utilities
Get the information about the current DataLoader worker process. |