Trait dfdx::tensor_ops::TryPool2D

pub trait TryPool2D<Kernel, Stride, Padding, Dilation>: Sized {
    type Pooled;
    type Error: Debug;

    // Required method
    fn try_pool2d(
        self,
        kind: Pool2DKind,
        kernel: Kernel,
        stride: Stride,
        padding: Padding,
        dilation: Dilation
    ) -> Result<Self::Pooled, Self::Error>;

    // Provided method
    fn pool2d(
        self,
        kind: Pool2DKind,
        kernel: Kernel,
        stride: Stride,
        padding: Padding,
        dilation: Dilation
    ) -> Self::Pooled { ... }
}

Required Associated Types

type Pooled

type Error: Debug

Required Methods

fn try_pool2d( self, kind: Pool2DKind, kernel: Kernel, stride: Stride, padding: Padding, dilation: Dilation ) -> Result<Self::Pooled, Self::Error>

Provided Methods

fn pool2d( self, kind: Pool2DKind, kernel: Kernel, stride: Stride, padding: Padding, dilation: Dilation ) -> Self::Pooled

Implementations on Foreign Types

impl<Kernel: Dim, Stride: Dim, Padding: Dim, Dilation: Dim> TryPool2D<Kernel, Stride, Padding, Dilation> for usize

type Pooled = usize

type Error = Infallible

fn try_pool2d( self, _: Pool2DKind, kernel: Kernel, stride: Stride, padding: Padding, dilation: Dilation ) -> Result<Self::Pooled, Self::Error>

Implementors

impl<Chan, Kernel, Stride, Padding, Dilation, Batch, H, W, E, D, T> TryPool2D<Kernel, Stride, Padding, Dilation> for Tensor<(Batch, Chan, H, W), E, D, T>where Chan: Dim, Kernel: Dim, Stride: Dim, Padding: Dim, Dilation: Dim, Batch: Dim, H: Dim + TryPool2D<Kernel, Stride, Padding, Dilation>, H::Pooled: Dim, W: Dim + TryPool2D<Kernel, Stride, Padding, Dilation>, W::Pooled: Dim, E: Dtype, D: Pool2DKernel<E>, T: Tape<E, D>,

type Pooled = Tensor<(Batch, Chan, <H as TryPool2D<Kernel, Stride, Padding, Dilation>>::Pooled, <W as TryPool2D<Kernel, Stride, Padding, Dilation>>::Pooled), E, D, T>

type Error = <D as HasErr>::Err

impl<Chan, Kernel, Stride, Padding, Dilation, H, W, E, D, T> TryPool2D<Kernel, Stride, Padding, Dilation> for Tensor<(Chan, H, W), E, D, T>where Chan: Dim, Kernel: Dim, Stride: Dim, Padding: Dim, Dilation: Dim, H: Dim + TryPool2D<Kernel, Stride, Padding, Dilation>, H::Pooled: Dim, W: Dim + TryPool2D<Kernel, Stride, Padding, Dilation>, W::Pooled: Dim, E: Dtype, D: Pool2DKernel<E> + ReshapeKernel<E>, T: Tape<E, D>,

type Pooled = Tensor<(Chan, <H as TryPool2D<Kernel, Stride, Padding, Dilation>>::Pooled, <W as TryPool2D<Kernel, Stride, Padding, Dilation>>::Pooled), E, D, T>

type Error = <D as HasErr>::Err

impl<const KERNEL: usize, const STRIDE: usize, const PADDING: usize, const DILATION: usize, const DIM: usize> TryPool2D<Const<KERNEL>, Const<STRIDE>, Const<PADDING>, Const<DILATION>> for Const<DIM>where Const<{ _ }>: Sized,

type Pooled = Const<{ (DIM + 2 * PADDING - DILATION * (KERNEL - 1) - 1) / STRIDE + 1 }>

type Error = Infallible