Module collate 

Collate - Batch Assembly Functions

Provides functions for combining individual samples into batches.

@version 0.1.0 @author AutomataNexus Development Team

Structs

DefaultCollate

Default collation strategy that stacks tensors.

StackCollate

Collation that stacks tensors along a new batch dimension.

Traits

Collate

Trait for collating samples into batches.

Functions

concat_tensors

Concatenates tensors along an existing dimension.

stack_tensors

Stacks a vector of tensors along dimension 0.