Trait Activate

pub trait Activate<T> {
    type Output;

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

The Activate 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> Activate<U> for Linear

type Output = U

impl<U> Activate<U> for ReLU

where U: ReLU,

type Output = <U as ReLU>::Output

impl<U> Activate<U> for Sigmoid

where U: Sigmoid,

type Output = <U as Sigmoid>::Output

impl<U> Activate<U> for Tanh

where U: Tanh,

type Output = <U as Tanh>::Output