fastnn 0.3.1

A small deep learning library in Rust with optional CUDA kernels
Documentation
1
2
3
4
5
6
7
8
9
//! Datasets, batching, and MNIST.

pub mod dataset;
pub mod loader;
pub mod mnist;

pub use dataset::{Dataset, TensorDataset};
pub use loader::{Batch, Batches, DataLoader};
pub use mnist::{Mnist, Split};