pub trait Dataset { // Required methods fn get(&self, index: usize) -> (Tensor, Tensor); fn len(&self) -> usize; // Provided method fn is_empty(&self) -> bool { ... } }
Generic dataset trait
Get a single sample
Get dataset size
Check if dataset is empty