Trait custos_math::nn::SoftmaxOps

pub trait SoftmaxOps<T> {
    fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>;
    fn softmax_grad(
        &self,
        activated: &Matrix<'_, T>,
        grads: &Matrix<'_, T>
    ) -> Matrix<'_, T>;
}

Required Methods

fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>

fn softmax_grad(
    &self,
    activated: &Matrix<'_, T>,
    grads: &Matrix<'_, T>
) -> Matrix<'_, T>

Implementations on Foreign Types

impl<T: Float + GenericBlas> SoftmaxOps<T> for CPU

fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>

fn softmax_grad(
    &self,
    activated: &Matrix<'_, T>,
    grads: &Matrix<'_, T>
) -> Matrix<'_, T>

impl<T: Default + Copy + GenericBlas> SoftmaxOps<T> for CudaDevice

fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>

fn softmax_grad(
    &self,
    activated: &Matrix<'_, T>,
    grads: &Matrix<'_, T>
) -> Matrix<'_, T>

impl<T: GenericBlas + Float> SoftmaxOps<T> for CLDevice

fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>

fn softmax_grad(
    &self,
    activated: &Matrix<'_, T>,
    grads: &Matrix<'_, T>
) -> Matrix<'_, T>

Implementors