blowtorch 0.1.0

Crate for running and creating Rust models trained in Pytorch.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

mod activation_functions;
mod layer_implementations;
mod linear;
mod flatten;
mod traits;
mod weight_loader;

pub use ndarray;

pub mod nn {
    pub use convolutions_rs::convolutions::ConvolutionLayer;
    pub use convolutions_rs::transposed_convolutions::TransposedConvolutionLayer;
    pub mod utils {
        pub use convolutions_rs::Padding;
    }
    pub mod loading {
        pub use crate::weight_loader::{NpzWeightLoader, WeightLoader};
    }
    pub use crate::activation_functions::ReluLayer;
    pub use crate::traits::{FloatLikePrimitive, Layer};
    pub use crate::flatten::Flatten;
    pub use crate::linear::LinearLayer;
}