Struct GlobalAvgPool 

pub struct GlobalAvgPool;

Global Average Pooling

Reduces spatial dimensions to 1x1 by averaging over all spatial positions. Commonly used before the final fully-connected layer in CNNs.

Input: [batch, height, width, channels] Output: [batch, 1, 1, channels]

Implementations

impl GlobalAvgPool

pub fn new() -> Self

Create a new global average pooling layer

Trait Implementations

impl Clone for GlobalAvgPool

fn clone(&self) -> GlobalAvgPool

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GlobalAvgPool

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

Formats the value using the given formatter.

impl Default for GlobalAvgPool

fn default() -> GlobalAvgPool

Returns the “default value” for a type.

impl Layer for GlobalAvgPool

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