Struct burn_dataset::transform::MapperDataset

pub struct MapperDataset<D, M, I> { /* private fields */ }

Dataset mapping each element in an inner dataset to another element type lazily.

Implementations

impl<D, M, I> MapperDataset<D, M, I>

pub fn new(dataset: D, mapper: M) -> Self

Constructs a new MapperDataset.

Trait Implementations

impl<D, M, I, O> Dataset<O> for MapperDataset<D, M, I>

where D: Dataset<I>, M: Mapper<I, O> + Send + Sync, I: Send + Sync, O: Send + Sync,

fn get(&self, index: usize) -> Option<O>

Gets the item at the given index.

fn len(&self) -> usize

Gets the number of items in the dataset.

fn is_empty(&self) -> bool

Checks if the dataset is empty.

fn iter(&self) -> DatasetIterator<'_, I>

where Self: Sized,

Returns an iterator over the dataset.