native_neural_network 0.1.6

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

use super::ActivationKind;

pub fn apply_activation_slice(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(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
}