Struct burn::nn::interpolate::Interpolate1dConfig

pub struct Interpolate1dConfig {
    pub output_size: Option<usize>,
    pub scale_factor: Option<f32>,
    pub mode: InterpolateMode,
}

Configuration for the 1D interpolation module.

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

Fields

output_size: Option<usize>

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

scale_factor: Option<f32>

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 Interpolate1dConfig

pub fn new() -> Interpolate1dConfig

Create a new instance of the config.

impl Interpolate1dConfig

pub fn with_output_size(self, output_size: Option<usize>) -> Interpolate1dConfig

Output size of the interpolated tensor.

pub fn with_scale_factor(self, scale_factor: Option<f32>) -> Interpolate1dConfig

Scale factor for resizing the input tensor.

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

Interpolation mode to use for resizing.

impl Interpolate1dConfig

pub fn init(self) -> Interpolate1d

Initialize the interpolation module

Trait Implementations

impl Clone for Interpolate1dConfig

fn clone(&self) -> Interpolate1dConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Config for Interpolate1dConfig

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 Interpolate1dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Interpolate1dConfig

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

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Interpolate1dConfig

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

Formats the value using the given formatter.

impl Serialize for Interpolate1dConfig

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.