Python PyTorch BatchSampler用法及代码示例

本文简要介绍python语言中 torch.utils.data.BatchSampler 的用法。

用法:

class torch.utils.data.BatchSampler(sampler, batch_size, drop_last)

参数：

• sampler(Sampler或者可迭代) -基础采样器。可以是任何可迭代对象

• batch_size(int) -小批量的大小。

• drop_last(bool) -如果 True ，如果其大小小于 batch_size，则采样器将丢弃最后一批

包装另一个采样器以产生小批量索引。

示例

>>> list(BatchSampler(SequentialSampler(range(10)), batch_size=3, drop_last=False))
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]
>>> list(BatchSampler(SequentialSampler(range(10)), batch_size=3, drop_last=True))
[[0, 1, 2], [3, 4, 5], [6, 7, 8]]