Trait ActivateGradient

pub trait ActivateGradient<T>: Activate<T> {
    type Input;
    type Delta;

    // Required method
    fn activate_gradient(&self, input: Self::Input) -> Self::Delta;
}

The ActivateGradient trait extends the Activate trait to include a method for computing the gradient of the activation function.

Required Associated Types

type Input

type Delta

Required Methods

fn activate_gradient(&self, input: Self::Input) -> Self::Delta

compute the gradient of some input

Implementors

impl<U> ActivateGradient<U> for Linear

where U: LinearActivation,

type Input = U

type Delta = <U as LinearActivation>::Output

impl<U> ActivateGradient<U> for ReLU

where U: ReLU,

type Input = U

type Delta = <U as ReLU>::Output

impl<U> ActivateGradient<U> for Sigmoid

where U: Sigmoid,

type Input = U

type Delta = <U as Sigmoid>::Output

impl<U> ActivateGradient<U> for Tanh

where U: Tanh,

type Input = U

type Delta = <U as Tanh>::Output