TensorPlay
API symbolsutils
Copy
View MarkdownDownload .md

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

tensorplay.use_deterministic_algorithms

Sets whether TensorPlay operations must use “deterministic” algorithms. That is, algorithms which, given the same input, and when run on the same software and hardware, always produce the same output. When enabled, opera

ChainDataset

Dataset for chaining multiple IterableDataset s.

Ask DeepWiki