Struct MaxPool2D 

pub struct MaxPool2D<F: Float + Debug + Send + Sync> { /* private fields */ }

2D Max Pooling layer

Implementations

impl<F: Float + Debug + Send + Sync + ScalarOperand + Default> MaxPool2D<F>

pub fn new( pool_size: (usize, usize), stride: (usize, usize), name: Option<&str>, ) -> Result<Self>

Trait Implementations

impl<F: Debug + Float + Debug + Send + Sync> Debug for MaxPool2D<F>

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

Formats the value using the given formatter.

impl<F: Float + Debug + Send + Sync + ScalarOperand + Default> Layer<F> for MaxPool2D<F>

fn forward(&self, input: &Array<F, IxDyn>) -> Result<Array<F, IxDyn>>

Forward pass of the layer

fn layer_type(&self) -> &str

Get the type of the layer (e.g., “Dense”, “Conv2D”)

fn inputshape(&self) -> Option<Vec<usize>>

Get the input shape if known

fn outputshape(&self) -> Option<Vec<usize>>

Get the output shape if known

fn name(&self) -> Option<&str>

Get the name of the layer if set

fn backward( &self, _input: &Array<F, IxDyn>, grad_output: &Array<F, IxDyn>, ) -> Result<Array<F, IxDyn>>

Backward pass of the layer to compute gradients

fn update(&mut self, _learningrate: F) -> Result<()>

Update the layer parameters with the given learning rate

fn as_any(&self) -> &dyn Any

Get the layer as a dyn Any for downcasting

fn as_any_mut(&mut self) -> &mut dyn Any

Get the layer as a mutable dyn Any for downcasting

fn params(&self) -> Vec<Array<F, IxDyn>>

Get the parameters of the layer

fn gradients(&self) -> Vec<Array<F, IxDyn>>

Get the gradients of the layer parameters

fn set_gradients(&mut self, _gradients: &[Array<F, IxDyn>]) -> Result<()>

Set the gradients of the layer parameters

fn set_params(&mut self, _params: &[Array<F, IxDyn>]) -> Result<()>

Set the parameters of the layer

fn set_training(&mut self, _training: bool)

Set the layer to training mode (true) or evaluation mode (false)

fn is_training(&self) -> bool

Get the current training mode

fn parameter_count(&self) -> usize

Get the number of trainable parameters in this layer

fn layer_description(&self) -> String

Get a detailed description of this layer