Struct DynamicBatchCollate 

pub struct DynamicBatchCollate<T: TensorElement> { /* private fields */ }

Dynamic batching collation for variable-size sequences

Implementations

impl<T: TensorElement> DynamicBatchCollate<T>

pub fn new(padding_value: T) -> Self

Create a new dynamic batch collation function

pub fn with_max_length(self, max_length: usize) -> Self

Set maximum sequence length (sequences longer than this will be truncated)

pub fn with_packing(self, pack: bool) -> Self

Enable sequence packing to minimize padding

Trait Implementations

impl<T: TensorElement + Copy + Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + Default> Collate<Tensor<T>> for DynamicBatchCollate<T>

type Output = (Tensor<T>, Tensor<i64>)

Output type after collation

fn collate(&self, batch: Vec<Tensor<T>>) -> Result<Self::Output>

Collate a batch of samples

fn expected_batch_size(&self) -> Option<usize>

Get the expected batch size (returns None for variable batch sizes)

fn supports_empty_batch(&self) -> bool

Check if this collate function supports empty batches

fn validate_batch(&self, batch: &[T]) -> Result<()>

Validate batch before collation (optional hook)