Enum ActivationFunction

pub enum ActivationFunction {
    ReLU,
    Sigmoid,
    Tanh,
    Softmax,
    LeakyReLU(f64),
    ELU(f64),
    GELU,
    Swish,
    Mish,
}

Serializable activation function

Variants

ReLU

ReLU activation

Sigmoid

Sigmoid activation

Tanh

Tanh activation

Softmax

Softmax activation

LeakyReLU(f64)

LeakyReLU activation

ELU(f64)

ELU activation

GELU

GELU activation

Swish

Swish activation

Mish

Mish activation

Implementations

impl ActivationFunction

pub fn from_name(name: &str) -> Option<Self>

Convert activation function name to ActivationFunction enum

pub fn to_name(&self) -> String

Convert ActivationFunction enum to activation function name

pub fn create<F: Float + Debug + ScalarOperand + Send + Sync>( &self, ) -> Box<dyn Activation<F>>

Create activation function from enum

Trait Implementations

impl Clone for ActivationFunction

fn clone(&self) -> ActivationFunction

Returns a duplicate of the value.

const fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ActivationFunction

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ActivationFunction

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ActivationFunction

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.