dataloader-rs 0.3.0

High-performance DataLoader for Rust with a PyTorch-like interface
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use crate::{collator::default_collator::DefaultCollate, error::Result};

macro_rules! primitive_impl {
    ($($t:ty),* $(,)?) => {
        $(
            impl DefaultCollate for $t {
                type Batch = Vec<$t>;
                fn collate_items(items: Vec<Self>) -> Result<Self::Batch> {
                    Ok(items)
                }
            }
        )*
    };
}

primitive_impl!(
    bool, char, u8, u16, u32, u64, u128, usize, i8, i16, i32, i64, i128, isize, f32, f64, String
);

#[cfg(test)]
mod tests {
    use crate::collator::{Collator, DefaultCollator};

    #[test]
    fn collates_primitives_to_vec() {
        let c = DefaultCollator;
        let out = c.collate(vec![1u32, 2, 3]).unwrap();
        assert_eq!(out, vec![1, 2, 3]);
    }
}