mlx_rs::nn

Type Alias NonLinearity

Source
pub type NonLinearity = dyn Fn(&Array, &Stream) -> Result<Array, Exception>;
Expand description

Type alias for the non-linearity function.