oarust 0.8.6

GPU-first semantic computing engine written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! Dataset values, batch iteration, and stateless dataset operations.
//!
//! Donor: `oa/data/dataset.h`, `oa/data/dsSubset.h`, `oa/data/fnDataset.h`
//!
//! | C++ | Rust |
//! | --- | --- |
//! | `oa::Dataset` | [`Dataset`] |
//! | `oa::Dataset::Sample` | [`Sample`] |
//! | `oa::DataLoader` | [`DataLoader`] |
//! | `oa::DataLoaderConfig` | [`DataLoaderConfig`] |
//! | `oa::DataLoader::Batch` | [`Batch`] |
//! | `oa::DsSubset` | [`Dataset::subset`] → [`Subset`] |
//! | `oa::FnDataset::SplitResult` | [`SplitResult`] |
//! | `oa::FnDataset::shuffle` | [`shuffle`] |
//! | `oa::FnDataset::randomSplit` | [`random_split`] |
//! | `oa::FnDataset::collate` | [`collate`] |

mod dataset;

pub use dataset::{
	Batch, DataLoader, DataLoaderConfig, Dataset, Sample, SplitResult, Subset, collate, random_split,
	shuffle,
};