Trait Activator

pub trait Activator<T> {
    type Output;

    // Required method
    fn activate(&self, input: T) -> Self::Output;
}

The Activator trait defines a method for applying an activation function to an input tensor.

Required Associated Types

type Output

Required Methods

fn activate(&self, input: T) -> Self::Output

Applies the activation function to the input tensor.

Implementors

impl<U> Activator<U> for Linear

where U: LinearActivation,

type Output = <U as LinearActivation>::Output

impl<U> Activator<U> for ReLU

where U: ReLU,

type Output = <U as ReLU>::Output

impl<U> Activator<U> for Sigmoid

where U: Sigmoid,

type Output = <U as Sigmoid>::Output

impl<U> Activator<U> for Tanh

where U: Tanh,

type Output = <U as Tanh>::Output

impl<U, V, F, S, D> Activator<U> for LayerBase<F, S, D>

where F: Activator<U, Output = V>, D: Dimension, S: RawData,

type Output = V

impl<X, Y, F> Activator<X> for F

where F: Fn(X) -> Y,

type Output = Y