Struct Activation 

pub struct Activation { /* private fields */ }

Generic activation layer that wraps an activation type.

Implementations

impl Activation

pub fn new(activation_type: ActivationType) -> Self

Creates a new activation layer.

pub fn activation_type(&self) -> ActivationType

Returns the activation type.

pub fn apply_inplace(&self, data: &mut [f32])

Applies the activation in-place.

impl Activation

pub fn relu() -> Self

Creates a ReLU activation.

pub fn relu6() -> Self

Creates a ReLU6 activation.

pub fn hard_swish() -> Self

Creates a HardSwish activation.

pub fn hard_sigmoid() -> Self

Creates a HardSigmoid activation (using Sigmoid as approximation).

pub fn identity() -> Self

Creates an identity (no-op) activation.

Trait Implementations

impl Clone for Activation

fn clone(&self) -> Activation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Activation

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

Formats the value using the given formatter.

impl Layer for Activation

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

Perform the forward pass

fn name(&self) -> &'static str

Get the layer name

fn num_params(&self) -> usize

Get the number of parameters