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

use ndarray_npy::ReadableElement;
use num_traits::{Float, FromPrimitive};
use std::{io::Read, ops::AddAssign};

/// Super trait that encompasses floats that behave like
/// primitives. This is used for all neural network
/// elements as basic operation type.
pub trait FloatLikePrimitive:
    'static + Float + AddAssign + FromPrimitive + ReadableElement
{
}

impl<T> FloatLikePrimitive for T where
    T: 'static + Float + AddAssign + FromPrimitive + ReadableElement
{
}

/// General model trait neural layers
pub trait Layer<I, O> {
    fn forward_pass(&self, input: &I) -> O;
}