Struct MaxPool1d 

pub struct MaxPool1d<D: Dtype, IT, OT, const RANK: usize> {
    pub kernel_l: usize,
    pub stride: usize,
    /* private fields */
}

1-D max pooling.

Fields

kernel_l: usize

The pooling window length.

stride: usize

The stride between pooling windows.

Implementations

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

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

Creates a new layer with the given kernel size and stride.

Trait Implementations

impl<D: Dtype, IT: Tensor<D, RANK> + EagerTensor, OT: Tensor<D, RANK>, const RANK: usize> Layer<IT> for MaxPool1d<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 MaxPool1d<D, SymTensor, SymTensor, RANK>

type Output = SymTensor

This layer’s output type.

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

Applies this layer to input.