Struct Mnist 

pub struct Mnist { /* private fields */ }

Implementations

impl Mnist

pub fn new() -> Mnist

pub fn load(path: &Path) -> Mnist

Trait Implementations

impl DataLoader for Mnist

fn get_size( &self, slice: Option<DataSlice>, ) -> Result<Vec<usize>, AutoDiffError>

The shape of the data if applicable.

fn get_item( &self, index: usize, slice: Option<DataSlice>, ) -> Result<(Var, Var), AutoDiffError>

Return one sample.

fn get_batch( &self, start: usize, end: usize, slice: Option<DataSlice>, ) -> Result<(Var, Var), AutoDiffError>

Return a batch following original order.

fn get_indexed_batch( &self, index: &[usize], slice: Option<DataSlice>, ) -> Result<(Var, Var), AutoDiffError>

Return a batch given the index.