pub trait Collate<T>: Send + Sync {
type Output;
// Required method
fn collate(&self, batch: Vec<T>) -> Self::Output;
}Expand description
Trait for collating samples into batches.
pub trait Collate<T>: Send + Sync {
type Output;
// Required method
fn collate(&self, batch: Vec<T>) -> Self::Output;
}Trait for collating samples into batches.