1 2 3 4 5 6
use crate::Tensor; pub trait Dataset { fn get(&self, idx: usize) -> (Tensor, Tensor); fn len(&self) -> usize; }