Trait LRN 

pub trait LRN<F>: NN<F> {
    // Required methods
    fn new_lrn_config(
        &self,
        n: u32,
        alpha: f64,
        beta: f64,
        k: f64,
    ) -> Result<Self::CLRN, Error>;
    fn lrn(
        &self,
        x: &SharedTensor<F>,
        result: &mut SharedTensor<F>,
        config: &Self::CLRN,
    ) -> Result<(), Error>;
    fn lrn_grad(
        &self,
        x: &SharedTensor<F>,
        x_diff: &SharedTensor<F>,
        result: &SharedTensor<F>,
        result_diff: &mut SharedTensor<F>,
        config: &Self::CLRN,
    ) -> Result<(), Error>;
}

Provides the functionality for a Backend to support Local Response Normalization operations.

Required Methods

fn new_lrn_config( &self, n: u32, alpha: f64, beta: f64, k: f64, ) -> Result<Self::CLRN, Error>

Creates a new (Local Response Normalization) LRNConfig, which needs to be passed to further LRN Operations.

fn lrn( &self, x: &SharedTensor<F>, result: &mut SharedTensor<F>, config: &Self::CLRN, ) -> Result<(), Error>

Computes a [LRN][lrn] over the input Tensor x. [lrn]: https://en.wikipedia.org/wiki/lrnal_neural_network

Saves the result to result.

fn lrn_grad( &self, x: &SharedTensor<F>, x_diff: &SharedTensor<F>, result: &SharedTensor<F>, result_diff: &mut SharedTensor<F>, config: &Self::CLRN, ) -> Result<(), Error>

Computes the gradient of a [LRN][lrn] over the input Tensor x. [lrn]: https://en.wikipedia.org/wiki/lrnal_neural_network

Saves the result to result_diff.

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.

Implementations on Foreign Types

impl<T> LRN<T> for Backend<Cuda>

where T: Float + Default + DataTypeInfo,

fn new_lrn_config( &self, n: u32, alpha: f64, beta: f64, k: f64, ) -> Result<Self::CLRN, Error>

fn lrn( &self, x: &SharedTensor<T>, result: &mut SharedTensor<T>, config: &Self::CLRN, ) -> Result<(), Error>

fn lrn_grad( &self, x: &SharedTensor<T>, x_diff: &SharedTensor<T>, result: &SharedTensor<T>, result_diff: &mut SharedTensor<T>, config: &Self::CLRN, ) -> Result<(), Error>

Implementors