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