Trait Convolution

pub trait Convolution
where
    Self: Sized,
{
    // Required methods
    fn conv2d(
        &self,
        filter: &Self,
        stride: (usize, usize),
        padding: (usize, usize),
        dilation: (usize, usize),
        padding_mode: PaddingMode,
    ) -> Self;
    fn conv2d_grad(
        &self,
        filter: &Self,
        stride: (usize, usize),
        padding: (usize, usize),
        dilation: (usize, usize),
        padding_mode: PaddingMode,
        output_grad: &Self,
    ) -> (Self, Self);
    fn conv_gen(
        &self,
        filter: &Self,
        stride: &[usize],
        padding: &[usize],
        dilation: &[usize],
        padding_mode: PaddingMode,
    ) -> Self;
    fn conv_grad_gen(
        &self,
        filter: &Self,
        stride: &[usize],
        padding: &[usize],
        dilation: &[usize],
        padding_mode: PaddingMode,
        output_grad: &Self,
    ) -> (Self, Self);
}

Required Methods

fn conv2d( &self, filter: &Self, stride: (usize, usize), padding: (usize, usize), dilation: (usize, usize), padding_mode: PaddingMode, ) -> Self

fn conv2d_grad( &self, filter: &Self, stride: (usize, usize), padding: (usize, usize), dilation: (usize, usize), padding_mode: PaddingMode, output_grad: &Self, ) -> (Self, Self)

fn conv_gen( &self, filter: &Self, stride: &[usize], padding: &[usize], dilation: &[usize], padding_mode: PaddingMode, ) -> Self

fn conv_grad_gen( &self, filter: &Self, stride: &[usize], padding: &[usize], dilation: &[usize], padding_mode: PaddingMode, output_grad: &Self, ) -> (Self, Self)

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.

Implementors

impl<T> Convolution for GenTensor<T>

where T: Float,