rotta_rs 0.0.5

a Deep Learning library with rust language
Documentation
1
2
3
4
5
6
use crate::Tensor;

pub trait Dataset {
    fn get(&self, idx: usize) -> (Tensor, Tensor);
    fn len(&self) -> usize;
}