1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
use libm::{
expf as libm_exp,
tanhf as libm_tanh,
fabsf as libm_abs,
sqrtf as libm_sqrt
};
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 + libm_exp(-x)) }
pub fn tanh(x: f32) -> f32 { libm_tanh(x) }
pub fn soft_sign(x: f32) -> f32 { x / (1.0 + libm_abs(x)) }
pub fn bent_identity(x: f32) -> f32 { ((libm_sqrt((x * x) + 1.0) - 1.0) / 2.0) + x }