Struct MaxPool2d

pub struct MaxPool2d { /* private fields */ }

Applies 2-dimensional max pooling.

The input is expected to be NHWC. The output will have the same N/C dimensions with the new H/W = floor((H/W - kernel)/stride) + 1

See MaxPool2d python docs for more information.

Implementations

impl MaxPool2d

pub fn new( kernel_size: impl Into<SingleOrPair<i32>>, stride: impl Into<SingleOrPair<usize>>, ) -> Self

Create a new 2-dimensional max pooling layer.

Params

• kernel_size: The size of the pooling window.
• stride: The stride of the pooling window.

Trait Implementations

impl Clone for MaxPool2d

fn clone(&self) -> MaxPool2d

Returns a copy 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 Module<&Array> for MaxPool2d

type Output = Array

Output type of the module.

type Error = Exception

Error type for the module.

fn forward(&mut self, x: &Array) -> Result<Array, Self::Error>

Forward pass of the module.

fn training_mode(&mut self, mode: bool)

Set whether the module is in training mode.

impl ModuleParameters for MaxPool2d

fn freeze_parameters(&mut self, recursive: bool)

Freeze all parameters in the module.

fn unfreeze_parameters(&mut self, recursive: bool)

Unfreeze all parameters in the module.

fn parameters(&self) -> ModuleParamRef<'_>

Get references to the module parameters.

fn parameters_mut(&mut self) -> ModuleParamMut<'_>

Get mutable references to the module parameters.

fn trainable_parameters(&self) -> ModuleParamRef<'_>

Get references to the trainable parameters. A parameter is trainable if it is NOT frozen.

fn all_frozen(&self) -> Option<bool>

Check if all parameters in the module are frozen. Returns None if there are no parameters.

fn any_frozen(&self) -> Option<bool>

Check if any parameter in the module is frozen. Returns None if there are no parameters.

fn update(&mut self, parameters: ModuleParam)

Update the module parameters.

fn update_flattened(&mut self, flattened_parameters: FlattenedModuleParam)

Update the module parameters from a flattened representation.