Trait custos_math::nn::SoftmaxOps

pub trait SoftmaxOps<T, D: Device = Self>: Device {
    // Required methods
    fn softmax(&self, inputs: &Matrix<'_, T, D>) -> Matrix<'_, T, Self>;
    fn softmax_grad(
        &self,
        activated: &Matrix<'_, T, D>,
        grads: &Matrix<'_, T, D>
    ) -> Matrix<'_, T, Self>;
}

Required Methods

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

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

Implementors

impl<T: Float + GenericBlas + MatrixMultiply> SoftmaxOps<T, CPU> for CPUwhere CPU: ColOp<T>,

impl<T: GenericBlas + MatrixMultiply + Float> SoftmaxOps<T, CUDA> for CUDA

impl<T: GenericBlas + MatrixMultiply + Float> SoftmaxOps<T, OpenCL> for OpenCL