mxnet.gluon.data.BatchSampler¶

class mxnet.gluon.data.BatchSampler(sampler, batch_size, last_batch='keep')[source]

Wraps over another Sampler and return mini-batches of samples.

Parameters: sampler (Sampler) – The source Sampler. batch_size (int) – Size of mini-batch. last_batch ({'keep', 'discard', 'rollover'}) – Specifies how the last batch is handled if batch_size does not evenly divide sequence length. If ‘keep’, the last batch will be returned directly, but will contain less element than batch_size requires. If ‘discard’, the last batch will be discarded. If ‘rollover’, the remaining elements will be rolled over to the next iteration.

Examples

>>> sampler = gluon.data.SequentialSampler(10)
>>> batch_sampler = gluon.data.BatchSampler(sampler, 3, 'keep')
>>> list(batch_sampler)
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

__init__(sampler, batch_size, last_batch='keep')[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

 __init__(sampler, batch_size[, last_batch]) Initialize self.