Struct MaxPool2dConfig

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

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.

Warning

Only symmetric padding is currently supported. As such, using Same padding with an even kernel size is not supported as it will not produce the same output size.

dilation: [usize; 2]

The dilation.

Implementations

impl MaxPool2dConfig

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

Create a new instance of the config.

impl MaxPool2dConfig

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

The strides.

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

The padding configuration.

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

The dilation.

impl MaxPool2dConfig

pub fn init(&self) -> MaxPool2d

Initialize a new max pool 2d module.

Trait Implementations

impl Clone for MaxPool2dConfig

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Config for MaxPool2dConfig

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

Available on crate feature std only.

Saves the configuration to a file.

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

Available on crate feature std only.

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

Formats the value using the given formatter.

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

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for MaxPool2dConfig

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

Formats the value using the given formatter.

impl Serialize for MaxPool2dConfig

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

where S: Serializer,

Serialize this value into the given Serde serializer.