Python mxnet.gluon.data.BatchSampler用法及代码示例

用法: class mxnet.gluon.data.BatchSampler(sampler, batch_size, last_batch='keep')

参数：

sampler：(Sampler) - 源采样器。
batch_size：(int) - 小批量的大小。
last_batch：({'keep', 'discard', 'rollover'}) -指定如果batch_size 不均分序列长度时如何处理最后一批。如果‘keep’，最后一批将直接返回，但包含的元素少于batch_size 所需的元素。如果‘discard’，最后一批将被丢弃。如果‘rollover’，剩余的元素将滚动到下一次迭代。

基础：mxnet.gluon.data.sampler.Sampler

包另一个 Sampler 并返回小批量样本。

例子：

>>> 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]]

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.gluon.data.BatchSampler。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。