Trait coaster_nn::Pooling

pub trait Pooling<F>: NN<F> {
    // Required methods
    fn new_pooling_config(
        &self,
        window: &[i32],
        stride: &[i32],
        padding: &[i32]
    ) -> Result<Self::CPOOL, Error>;
    fn pooling_max(
        &self,
        x: &SharedTensor<F>,
        result: &mut SharedTensor<F>,
        config: &Self::CPOOL
    ) -> Result<(), Error>;
    fn pooling_max_grad(
        &self,
        x: &SharedTensor<F>,
        x_diff: &SharedTensor<F>,
        result: &SharedTensor<F>,
        result_diff: &mut SharedTensor<F>,
        config: &Self::CPOOL
    ) -> Result<(), Error>;
    fn pooling_avg(
        &self,
        x: &SharedTensor<F>,
        result: &mut SharedTensor<F>,
        config: &Self::CPOOL
    ) -> Result<(), Error>;
    fn pooling_avg_grad(
        &self,
        x: &SharedTensor<F>,
        x_diff: &SharedTensor<F>,
        result: &SharedTensor<F>,
        result_diff: &mut SharedTensor<F>,
        config: &Self::CPOOL
    ) -> Result<(), Error>;
}

Provides the functionality for a Backend to support Pooling operations.

Required Methods

fn new_pooling_config( &self, window: &[i32], stride: &[i32], padding: &[i32] ) -> Result<Self::CPOOL, Error>

Creates a new PoolingConfig, which needs to be passed to further pooling Operations.

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

Computes non-linear down-sampling ([max Pooling][pooling]) over the input Tensor x. [pooling]: https://en.wikipedia.org/wiki/Convolutional_neural_network#Pooling_layer

Saves the result to result.

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

Computes the gradient of [max Pooling][pooling] over the input Tensor x. [pooling]: https://en.wikipedia.org/wiki/Convolutional_neural_network#Pooling_layer

Saves the result to result_diff.

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

Computes non-linear down-sampling ([average Pooling][pooling]) over the input Tensor x. [pooling]: https://en.wikipedia.org/wiki/Convolutional_neural_network#Pooling_layer

Saves the result to result.

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

Computes the gradient of [average Pooling][pooling] over the input Tensor x. [pooling]: https://en.wikipedia.org/wiki/Convolutional_neural_network#Pooling_layer

Saves the result to result_diff.

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl<T> Pooling<T> for Backend<Cuda>

where T: Float + Default + DataTypeInfo,

fn new_pooling_config( &self, window: &[i32], stride: &[i32], padding: &[i32] ) -> Result<Self::CPOOL, Error>

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

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

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

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

impl<T> Pooling<T> for Backend<Native>

where T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy + PartialOrd + Bounded,

fn new_pooling_config( &self, window: &[i32], stride: &[i32], padding: &[i32] ) -> Result<Self::CPOOL, Error>

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

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

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

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

Implementors