kratom 1.0.1

Simple neural network library.
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

use super::Multiplier;



pub struct RegularMultiplier {}
impl RegularMultiplier {
    #[inline]
    pub fn new() -> Self {
        RegularMultiplier { }
    }
}
impl Multiplier for RegularMultiplier {
    #[inline]
    fn multiply(&mut self, a: &[f32], b: &[f32], sizes: (usize, usize, usize), c: &mut [f32]) {
        multiply(a, b, sizes, c)
    }
}


/// Standard matrix multiplication
pub fn multiply(a: &[f32], b: &[f32], sizes: (usize, usize, usize), c: &mut [f32]) {
    // note: sizes = (A height, A width && B height, B width)
    let (m, n, p) = sizes;

    for i in 0..m { //A height
        for j in 0..p { //B width
            // note: chooses a row in A and a column in B and sums the products
            let mut sum = 0.0;
            for k in 0..n { //A width and B height
                sum += a[i*n + k] * b[k*p + j];
            }
            c[i*p + j] = sum;
        }
    }
}