Trait coaster_nn::LogSoftmax

pub trait LogSoftmax<F>: NN<F> {
    // Required methods
    fn log_softmax(
        &self,
        x: &SharedTensor<F>,
        result: &mut SharedTensor<F>
    ) -> Result<(), Error>;
    fn log_softmax_grad(
        &self,
        x: &SharedTensor<F>,
        x_diff: &SharedTensor<F>,
        result_diff: &mut SharedTensor<F>
    ) -> Result<(), Error>;
}

Provides the functionality for a Backend to support LogSoftmax operations.

Required Methods

fn log_softmax( &self, x: &SharedTensor<F>, result: &mut SharedTensor<F> ) -> Result<(), Error>

Computes a logarithmic softmax over the input Tensor x.

Saves the result to result.

fn log_softmax_grad( &self, x: &SharedTensor<F>, x_diff: &SharedTensor<F>, result_diff: &mut SharedTensor<F> ) -> Result<(), Error>

Computes the gradient of a logarithmic softmax over the input Tensor x.

Saves the result to result_diff.

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl LogSoftmax<f32> for Backend<Native>

fn log_softmax( &self, x: &SharedTensor<f32>, result: &mut SharedTensor<f32> ) -> Result<(), Error>

fn log_softmax_grad( &self, x: &SharedTensor<f32>, x_diff: &SharedTensor<f32>, result_diff: &mut SharedTensor<f32> ) -> Result<(), Error>

impl LogSoftmax<f64> for Backend<Native>

fn log_softmax( &self, x: &SharedTensor<f64>, result: &mut SharedTensor<f64> ) -> Result<(), Error>

fn log_softmax_grad( &self, x: &SharedTensor<f64>, x_diff: &SharedTensor<f64>, result_diff: &mut SharedTensor<f64> ) -> Result<(), Error>

impl<T> LogSoftmax<T> for Backend<Cuda>

where T: Float + Default + DataTypeInfo,

fn log_softmax( &self, x: &SharedTensor<T>, result: &mut SharedTensor<T> ) -> Result<(), Error>

fn log_softmax_grad( &self, x: &SharedTensor<T>, x_diff: &SharedTensor<T>, result_diff: &mut SharedTensor<T> ) -> Result<(), Error>

Implementors