Struct SELU 

pub struct SELU { /* private fields */ }

SELU (Scaled Exponential Linear Unit) activation module

Implementations

impl SELU

pub fn new() -> Self

pub fn with_params(alpha: f32, scale: f32) -> Self

Trait Implementations

impl Default for SELU

fn default() -> Self

Returns the “default value” for a type.

impl Module for SELU

fn forward(&self, input: &Tensor) -> Tensor

Forward pass

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

fn train(&mut self)

Set module to training mode

fn eval(&mut self)

Set module to evaluation mode

fn is_training(&self) -> bool

Check if in training mode

fn num_parameters(&self) -> usize

Number of trainable parameters