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

use crate::initializers::InitKind;

pub fn layers_are_valid(layers: &[usize]) -> bool {
    if layers.len() < 2 {
        return false;
    }
    for &dim in layers {
        if dim == 0 {
            return false;
        }
    }
    true
}

pub fn init_kind_is_finite(kind: InitKind) -> bool {
    match kind {
        InitKind::Constant(v) => v.is_finite(),
        _ => true,
    }
}

pub fn params_all_finite(params: &[f32]) -> bool {
    for &v in params {
        if !v.is_finite() {
            return false;
        }
    }
    true
}