Activation

scirs2_neural::activations_minimal

Trait Activation 

Source 
pub trait Activation<F> {
    // Required methods
    fn forward(&self, input: &Array<F, IxDyn>) -> Result<Array<F, IxDyn>>;
    fn backward(
        &self,
        grad_output: &Array<F, IxDyn>,
        input: &Array<F, IxDyn>,
    ) -> Result<Array<F, IxDyn>>;
}
Expand description

Trait for activation functions

Required Methods§

Source

fn forward(&self, input: &Array<F, IxDyn>) -> Result<Array<F, IxDyn>>

Forward pass of the activation function

Source

fn backward( &self, grad_output: &Array<F, IxDyn>, input: &Array<F, IxDyn>, ) -> Result<Array<F, IxDyn>>

Backward pass of the activation function

Implementors§