BatchSampler

class tensorplay.utils.data.BatchSampler(sampler: Sampler[int] | Iterable[int], batch_size: int, drop_last: bool)[source]

Wraps another sampler to yield a mini-batch of indices.

Parameters:

• sampler (Sampler or Iterable) – Base sampler. Can be any iterable object

• batch_size (int) – Size of mini-batch.

• drop_last (bool) – If True, the sampler will drop the last batch if its size would be less than batch_size