rust_ml/core/activations/
activation.rs

1use ndarray::{Array, Dimension};
2
3pub trait Activation<D>
4where
5    D: Dimension,
6{
7    fn activate(z: &Array<f64, D>) -> Array<f64, D>;
8    fn derivative(z: &Array<f64, D>) -> Array<f64, D>;
9}