Enum burn_core::nn::conv::Conv2dPaddingConfig

pub enum Conv2dPaddingConfig {
    Same,
    Valid,
    Explicit(usize, usize),
}

Padding configuration for 2D convolution config.

Variants

Same

Dynamicaly calculate the amount of padding necessary to ensure that the output size will be the same as the input.

Valid

Same as no padding.

Explicit(usize, usize)

Applies the specified amount of padding to all inputs.

Trait Implementations

impl<B: ADBackend> ADModule<B> for Conv2dPaddingConfig

type InnerModule = Conv2dPaddingConfig

fn valid(&self) -> Self::InnerModule

Get the same module, but on the inner backend without auto-differentiation.

impl Clone for Conv2dPaddingConfig

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 Conv2dPaddingConfig

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

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

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

impl Debug for Conv2dPaddingConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Conv2dPaddingConfig

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

Deserialize this value from the given Serde deserializer.

impl Display for Conv2dPaddingConfig

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

Formats the value using the given formatter.

impl<B: Backend> Module<B> for Conv2dPaddingConfig

type Record = ()

Type to save and load the module.

fn visit<V: ModuleVisitor<B>>(&self, _visitor: &mut V)

Visit each tensor in the module with a visitor.

fn map<M: ModuleMapper<B>>(self, _mapper: &mut M) -> Self

Map each tensor in the module with a mapper.

fn load_record(self, _record: Self::Record) -> Self

Load the module state from a record.

fn into_record(self) -> Self::Record

Convert the module into a record containing the state.

fn devices(&self) -> Vec<B::Device>

Get the device list of the module and all of its sub-modules.

fn fork(self, device: &B::Device) -> Self

Fork the module and all of its sub-modules to the given device.

fn to_device(self, device: &B::Device) -> Self

Move the module and all of its sub-modules to the given device.

fn no_grad(self) -> Self

Each tensor in the module tree will not require grad.

fn num_params(&self) -> usize

Get the number of parameters the module has, including all of its sub-modules.

impl Serialize for Conv2dPaddingConfig

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

Serialize this value into the given Serde serializer.