Struct SimpleConfig 

pub struct SimpleConfig {
    pub batch_size: usize,
    pub shuffle: bool,
    pub num_workers: usize,
    pub drop_last: bool,
    pub generator: Option<u64>,
}

Create a simple DataLoader with automatic sampling strategy

This function automatically chooses between sequential and random sampling based on the shuffle parameter.

Arguments

• dataset - The dataset to iterate over
• batch_size - Number of samples per batch
• shuffle - Whether to use random sampling (true) or sequential sampling (false)
• generator - Optional random seed for reproducible shuffling (only used when shuffle=true)

Returns

Either a SimpleDataLoader or SimpleRandomDataLoader depending on shuffle setting

Examples

use torsh_data::dataloader::simple::{simple_dataloader, simple_random_dataloader};
use torsh_data::dataset::TensorDataset;

let dataset = TensorDataset::new(vec![1, 2, 3, 4, 5]);

// Sequential sampling
let sequential_loader = simple_dataloader(dataset, 2, false)?;
for batch in sequential_loader.iter() {
    // Process batch
}

Configuration for simple DataLoader creation

This struct provides a more structured way to configure simple DataLoaders while maintaining the convenience of the simple API.

Fields

batch_size: usize

Number of samples per batch

shuffle: bool

Whether to shuffle the data

num_workers: usize

Number of worker threads (0 for single-threaded)

drop_last: bool

Whether to drop the last incomplete batch

generator: Option<u64>

Optional random seed for reproducible results

Implementations

impl SimpleConfig

pub fn new() -> Self

Create a new simple configuration

pub fn batch_size(self, batch_size: usize) -> Self

Set batch size

pub fn shuffle(self, shuffle: bool) -> Self

Set shuffle

pub fn num_workers(self, num_workers: usize) -> Self

Set number of workers

pub fn drop_last(self, drop_last: bool) -> Self

Set drop last

pub fn generator(self, seed: u64) -> Self

Set random generator seed

Trait Implementations

impl Clone for SimpleConfig

fn clone(&self) -> SimpleConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SimpleConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for SimpleConfig

fn default() -> Self

Returns the “default value” for a type.