Trait auto_diff_data_pipe::dataloader::DataLoader

pub trait DataLoader {
    fn get_size(
        &self, 
        slice: Option<DataSlice>
    ) -> Result<Vec<usize>, AutoDiffError>;
    fn get_item(
        &self, 
        index: usize, 
        slice: Option<DataSlice>
    ) -> Result<(Var, Var), AutoDiffError>;
    fn get_batch(
        &self, 
        start: usize, 
        end: usize, 
        slice: Option<DataSlice>
    ) -> Result<(Var, Var), AutoDiffError>;

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

Required methods

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.

Provided methods

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

Return a batch given the index.

Implementors

impl DataLoader for Mnist