matrux 0.1.0

Neural network implementation in rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use crate::{Scalar, MatrixPlan, Matrix};


#[derive(Clone, Debug)]
pub enum MatrixOp<I: Scalar> {
    Input {
        name: String,
    },
    Output {
        name: String,
        matrix: MatrixPlan<I>,
    },
    Constant {
        matrix: Matrix<I>,
    },
    Scale {
        matrix: MatrixPlan<I>,
        scalar: I,
    },
    Max {
        matrix: MatrixPlan<I>,
        scalar: I,
    },
    Neg {
        matrix: MatrixPlan<I>,
    },
    Transpose {
        matrix: MatrixPlan<I>,
    },
    Sign {
        matrix: MatrixPlan<I>,
    },
    Sigmoid {
        matrix: MatrixPlan<I>,
    },
    Mul {
        left: MatrixPlan<I>,
        right: MatrixPlan<I>,
    },
    HadamardMul {
        left: MatrixPlan<I>,
        right: MatrixPlan<I>,
    },
    Add {
        left: MatrixPlan<I>,
        right: MatrixPlan<I>,
    },
    Sub {
        left: MatrixPlan<I>,
        right: MatrixPlan<I>,
    },
    Combine {
        inner: Vec<MatrixPlan<I>>,
    },
}