Struct LpPool3d 

pub struct LpPool3d<D: Dtype, IT, OT, const RANK: usize> {
    pub kernel_d: usize,
    pub kernel_h: usize,
    pub kernel_w: usize,
    pub stride_d: usize,
    pub stride_h: usize,
    pub stride_w: usize,
    pub p: f64,
    /* private fields */
}

3-D power-average (Lp) pooling: like average pooling, generalized to an arbitrary p-norm.

Fields

kernel_d: usize

Pooling window depth.

kernel_h: usize

Pooling window height.

kernel_w: usize

Pooling window width.

stride_d: usize

Stride along the depth dimension.

stride_h: usize

Stride along the height dimension.

stride_w: usize

Stride along the width dimension.

p: f64

The p in the p-norm.

Implementations

impl<D: Dtype, IT, OT, const RANK: usize> LpPool3d<D, IT, OT, RANK>

pub fn new( kernel_size: (usize, usize, usize), stride: (usize, usize, usize), p: f64, ) -> Self

Creates a new layer with the given kernel size, stride, and p-norm.

Trait Implementations

impl<D: Dtype, IT: Tensor<D, RANK> + EagerTensor, OT: Tensor<D, RANK>, const RANK: usize> Layer<IT> for LpPool3d<D, IT, OT, RANK>

type Output = OT

This layer’s output type.

fn call(&self, _input: IT) -> Self::Output

Applies this layer to input.

impl<D: Dtype, const RANK: usize> Layer<SymTensor> for LpPool3d<D, SymTensor, SymTensor, RANK>

type Output = SymTensor

This layer’s output type.

fn call(&self, input: SymTensor) -> SymTensor

Applies this layer to input.