const_cge 0.2.1

Neural network compiler with a focus on no_std targets
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

#[allow(unused_imports)]
use micromath::F32Ext;
use libm::tanhf as libm_tanh;

pub const fn linear(x: f32)        -> f32 { x }
pub       fn threshold(x: f32)     -> f32 { if x > 0.0 { 1.0 } else { 0.0 } }
pub       fn relu(x: f32)          -> f32 { if x > 0.0 { x } else { 0.0 } } 
pub       fn sign(x: f32)          -> f32 { if x > 0.0 { 1.0 } else if x == 0.0 { 0.0 } else { -1.0 } }
pub       fn sigmoid(x: f32)       -> f32 { 1.0 / (1.0 + (-x).exp()) }
pub       fn tanh(x: f32)          -> f32 { libm_tanh(x) }
pub       fn soft_sign(x: f32)     -> f32 { x / (1.0 + x.abs()) }
pub       fn bent_identity(x: f32) -> f32 { ((((x * x) + 1.0).sqrt() - 1.0) / 2.0) + x }