1 2 3 4 5 6 7 8
pub mod dataloader; pub use dataloader::DataLoader; use lumen_core::Tensor; pub trait DataSet { fn len(&self) -> usize; fn get(&self, index: usize) -> (Tensor, Tensor); }