Struct auto_diff_data_pipe::dataloader::mnist::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.