mindspore.dataset.dataloader.BatchSampler
- class mindspore.dataset.dataloader.BatchSampler(sampler, batch_size, drop_last)[source]
Sampler that yields a mini-batch of indices each time.
- Parameters
Examples
>>> from mindspore.dataset.dataloader import BatchSampler, SequentialSampler >>> >>> dataset = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] >>> sequential_sampler = SequentialSampler(dataset) >>> >>> batch_sampler = BatchSampler(sequential_sampler, 4, False) >>> print(list(batch_sampler)) [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]] >>> >>> batch_sampler = BatchSampler(sequential_sampler, 4, True) >>> print(list(batch_sampler)) [[0, 1, 2, 3], [4, 5, 6, 7]]