native_neural_network 0.3.1

Lib no_std Rust for native neural network (.rnn)
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

use super::ActivationKind;

pub fn apply_activation_slice_f32(kind: ActivationKind, input: &[f32], out: &mut [f32]) -> bool {
    if out.len() < input.len() {
        return false;
    }
    for i in 0..input.len() {
        out[i] = kind.apply(input[i]);
    }
    true
}

pub fn activation_lut_f32(kind: ActivationKind, start: f32, step: f32, out: &mut [f32]) -> bool {
    if !start.is_finite() || !step.is_finite() {
        return false;
    }
    for (i, v) in out.iter_mut().enumerate() {
        *v = kind.apply(start + step * i as f32);
    }
    true
}

pub fn apply_activation_slice_f64(kind: ActivationKind, input: &[f64], out: &mut [f64]) -> bool {
    if out.len() < input.len() {
        return false;
    }
    for i in 0..input.len() {
        out[i] = kind.apply_f64(input[i]);
    }
    true
}

pub fn activation_lut_f64(kind: ActivationKind, start: f64, step: f64, out: &mut [f64]) -> bool {
    if !start.is_finite() || !step.is_finite() {
        return false;
    }
    for (i, v) in out.iter_mut().enumerate() {
        *v = kind.apply_f64(start + step * i as f64);
    }
    true
}