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
10
11
12
13
14
15
//! Tensors: shape, storage and the link into the autograd graph.

mod checked;
mod core;
mod device;
mod display;
mod init;
pub mod ops;
pub mod shape;
pub mod storage;

pub use core::Tensor;
pub use device::Device;
pub use ops::conv::Window;
pub use ops::norm::BatchStats;