Struct Softmax 

pub struct Softmax { /* private fields */ }

Softmax activation function

Implementations

impl Softmax

pub fn new(axis: isize) -> Softmax

Trait Implementations

impl<F> Activation<F> for Softmax

where F: Float + Debug,

fn forward( &self, input: &ArrayBase<OwnedRepr<F>, Dim<IxDynImpl>>, ) -> Result<ArrayBase<OwnedRepr<F>, Dim<IxDynImpl>>, NeuralError>

Forward pass of the activation function

fn backward( &self, grad_output: &ArrayBase<OwnedRepr<F>, Dim<IxDynImpl>>, input: &ArrayBase<OwnedRepr<F>, Dim<IxDynImpl>>, ) -> Result<ArrayBase<OwnedRepr<F>, Dim<IxDynImpl>>, NeuralError>

Backward pass of the activation function

impl Clone for Softmax

fn clone(&self) -> Softmax

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for Softmax

impl Debug for Softmax

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

Formats the value using the given formatter.

impl Default for Softmax

fn default() -> Softmax

Returns the “default value” for a type.