Struct MaxPool2dConfig 

pub struct MaxPool2dConfig {
    pub kernel_size: [usize; 2],
    pub strides: [usize; 2],
    pub padding: PaddingConfig2d,
    pub dilation: [usize; 2],
    pub ceil_mode: bool,
}

Configuration to create a 2D max pooling layer using the init function.

Fields

kernel_size: [usize; 2]

The size of the kernel.

strides: [usize; 2]

The strides.

padding: PaddingConfig2d

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

dilation: [usize; 2]

The dilation.

ceil_mode: bool

If true, use ceiling instead of floor for output size calculation.

Implementations

impl MaxPool2dConfig

pub fn new(kernel_size: [usize; 2]) -> MaxPool2dConfig

Create a new instance of the config.

Arguments

Required Arguments

kernel_size

The size of the kernel.

Default Arguments

strides

The strides.

• Defaults to "kernel_size"

padding

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

• Defaults to "PaddingConfig2d::Valid"

dilation

The dilation.

• Defaults to "[1, 1]"

ceil_mode

If true, use ceiling instead of floor for output size calculation.

• Defaults to "false"

impl MaxPool2dConfig

pub fn with_strides(self, strides: [usize; 2]) -> MaxPool2dConfig

Sets the value for the field strides.

The strides.

• Defaults to "kernel_size"

pub fn with_padding(self, padding: PaddingConfig2d) -> MaxPool2dConfig

Sets the value for the field padding.

The padding configuration.

Supports symmetric and asymmetric padding. Same padding with even kernel sizes will automatically use asymmetric padding to preserve input dimensions.

• Defaults to "PaddingConfig2d::Valid"

pub fn with_dilation(self, dilation: [usize; 2]) -> MaxPool2dConfig

Sets the value for the field dilation.

The dilation.

• Defaults to "[1, 1]"

pub fn with_ceil_mode(self, ceil_mode: bool) -> MaxPool2dConfig

Sets the value for the field ceil_mode.

If true, use ceiling instead of floor for output size calculation.

• Defaults to "false"

impl MaxPool2dConfig

pub fn init(&self) -> MaxPool2d

Initialize a new max pool 2d module.

Trait Implementations

impl Clone for MaxPool2dConfig

fn clone(&self) -> MaxPool2dConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for MaxPool2dConfig

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for MaxPool2dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MaxPool2dConfig

fn deserialize<D>( deserializer: D, ) -> Result<MaxPool2dConfig, <D as Deserializer<'de>>::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for MaxPool2dConfig

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

Formats the value using the given formatter.

impl Serialize for MaxPool2dConfig

fn serialize<S>( &self, serializer: S, ) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.