Struct burn::nn::interpolate::Interpolate2dConfig

pub struct Interpolate2dConfig {
    pub output_size: Option<[usize; 2]>,
    pub scale_factor: Option<[f32; 2]>,
    pub mode: InterpolateMode,
}

Configuration for the 2D interpolation module.

This struct defines the configuration options for the 2D interpolation operation. It allows specifying the output size, scale factor, and interpolation mode.

Fields

output_size: Option<[usize; 2]>

Output size of the interpolated tensor. If specified, this takes precedence over scale_factor.

scale_factor: Option<[f32; 2]>

Scale factor for resizing the input tensor. This is used when output_size is not specified.

mode: InterpolateMode

Interpolation mode to use for resizing. Determines how the output values are calculated.

Implementations

impl Interpolate2dConfig

pub fn new() -> Interpolate2dConfig

Create a new instance of the config.

impl Interpolate2dConfig

pub fn with_output_size( self, output_size: Option<[usize; 2]>, ) -> Interpolate2dConfig

Output size of the interpolated tensor.

pub fn with_scale_factor( self, scale_factor: Option<[f32; 2]>, ) -> Interpolate2dConfig

Scale factor for resizing the input tensor.

pub fn with_mode(self, mode: InterpolateMode) -> Interpolate2dConfig

Interpolation mode to use for resizing.

impl Interpolate2dConfig

pub fn init(self) -> Interpolate2d

Initialize the interpolation module

Trait Implementations

impl Clone for Interpolate2dConfig

fn clone(&self) -> Interpolate2dConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Config for Interpolate2dConfig

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 Interpolate2dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Interpolate2dConfig

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

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Interpolate2dConfig

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

Formats the value using the given formatter.

impl Serialize for Interpolate2dConfig

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.