Struct DataLoader 

pub struct DataLoader { /* private fields */ }

Parallel data loader that prefetches batches using worker threads.

Implementations

impl DataLoader

pub fn new( inputs: Vec<Tensor>, targets: Vec<Tensor>, config: DataLoaderConfig, ) -> Result<Self, ModelError>

Creates a new data loader from individual sample tensors and configuration.

All input tensors must have the same shape, and all target tensors must have the same shape. The number of inputs must equal the number of targets.

pub fn len(&self) -> usize

Returns the number of batches per epoch.

pub fn is_empty(&self) -> bool

Returns true if the loader has no batches.

pub fn config(&self) -> &DataLoaderConfig

Returns the underlying configuration.

pub fn sample_count(&self) -> usize

Returns the total number of samples.

pub fn iter(&self) -> DataLoaderIter

Creates an iterator that spawns worker threads and prefetches batches.

Each call increments the internal epoch counter, producing a different shuffle order when config.shuffle is true.