Trait Activation 

pub trait Activation<B: Backend>: Clone {
    // Required methods
    fn activate<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>;
    fn derivative<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>;

    // Provided method
    fn vjp<D: Dimension>(
        &self,
        z: &B::Tensor<D>,
        grad: &B::Tensor<D>,
    ) -> B::Tensor<D> { ... }
}

Required Methods

fn activate<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>

fn derivative<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>

Provided Methods

fn vjp<D: Dimension>( &self, z: &B::Tensor<D>, grad: &B::Tensor<D>, ) -> B::Tensor<D>

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<B: Backend> Activation<B> for ReLU

impl<B: Backend> Activation<B> for Sigmoid

impl<B: Backend> Activation<B> for Softmax

impl<B: Backend> Activation<B> for Tanh