rumus 0.3.1

A native-Rust deep learning framework with explicit memory safety and hardware acceleration
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
31

// SPDX-License-Identifier: Apache-2.0 OR MIT
//! Dataset trait and data item types.

use crate::tensor::Tensor;

/// A single data item: input-target pair.
pub struct DataItem {
    pub input: Tensor,
    pub target: Tensor,
}

/// Trait for indexable datasets.
///
/// Implementations must be `Send + Sync` because `DataLoader` worker threads
/// call `get()` concurrently via `Arc<dyn Dataset>`.
pub trait Dataset: Send + Sync {
    /// Total number of items in the dataset.
    fn len(&self) -> usize;

    /// Retrieve the item at `index`.
    ///
    /// # Panics
    ///
    /// May panic if `index >= len()`.
    fn get(&self, index: usize) -> DataItem;

    /// Whether the dataset is empty.
    fn is_empty(&self) -> bool {
        self.len() == 0
    }
}