Module ai_dataloader::collate

Merges a list of samples to form a batch.

Structs

• DefaultCollate
  Default Collate function that mimic the default_collate function from PyTorch.
• NoOpCollate
  Simple Collate that doesn’t change the batch of samples.
• TorchCollatetch
  Torch Collate function that mimic the default_collate function from PyTorch.

Traits

• Collate
  Any collate gather samples from one batch together.