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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。