Struct MaxPool2d 

pub struct MaxPool2d { /* private fields */ }

2D Max Pooling

Performs max pooling over spatial dimensions with configurable kernel size, stride, and padding.

Implementations

impl MaxPool2d

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

Create a new max pooling layer

pub fn with_kernel(kernel_size: usize) -> Self

Create a max pooling layer with stride equal to kernel size

pub fn output_shape(&self, input_shape: &[usize]) -> CnnResult<Vec<usize>>

Get the output shape for a given input shape

Trait Implementations

impl Clone for MaxPool2d

fn clone(&self) -> MaxPool2d

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MaxPool2d

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Layer for MaxPool2d

fn forward(&self, input: &Tensor) -> CnnResult<Tensor>

Perform the forward pass

fn name(&self) -> &'static str

Get the layer name

fn num_params(&self) -> usize

Get the number of parameters