axtreme.data.multi_dim_batch_sampler

Allows retuning batches of data of an arbitrary dimension.

Classes

MultiBatchSampler(sampler, batch_shape[, ...])

Reads the entire sampler into batches of shape batch_shape.