Struct AvgPool2d 

pub struct AvgPool2d { /* private fields */ }

2D Average Pooling

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

Implementations

impl AvgPool2d

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

Create a new average pooling layer

pub fn with_kernel(kernel_size: usize) -> Self

Create an average 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 AvgPool2d

fn clone(&self) -> AvgPool2d

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AvgPool2d

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

Formats the value using the given formatter.

impl Layer for AvgPool2d

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