Struct BatchSampler 

pub struct BatchSampler<S: Sampler> { /* private fields */ }

Batches indices produced by an inner Sampler.

Implementations

impl<S: Sampler> BatchSampler<S>

pub fn new(sampler: S, batch_size: usize, drop_last: bool) -> Self

Build a batch sampler from an index sampler.

pub fn batch_size(&self) -> usize

pub fn drop_last(&self) -> bool

pub fn set_batch_size(&mut self, batch_size: usize)

pub fn set_drop_last(&mut self, drop_last: bool)

pub fn len(&self, dataset_len: usize) -> usize

Number of batches in one epoch, without advancing the sampler.

pub fn num_batches(&self, n_indices: usize) -> usize

Number of batches n_indices sampled indices are grouped into.

pub fn batch_indices(&mut self, dataset_len: usize) -> Vec<Vec<usize>>

Return one epoch of grouped batch indices.