Struct InterpolateOptions 

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

Interpolation options.

Fields

mode: InterpolateMode

Algorithm used.

align_corners: bool

If true, the input and output tensors are aligned by their corner pixels. If false, half-pixel coordinate mapping is used instead.

output_size: Option<[usize; 2]>

Output spatial size [height, width].

Exactly one of output_size or scale_factor must be set when calling burn_tensor::module::interpolate. Backends receive the resolved size as a separate argument and ignore this field.

scale_factor: Option<[f32; 2]>

Scale factor [scale_height, scale_width] applied to the input spatial size. The output size is floor(input_size * scale_factor).

Exactly one of output_size or scale_factor must be set when calling burn_tensor::module::interpolate. Backends ignore this field.

Implementations

impl InterpolateOptions

pub fn new(mode: InterpolateMode) -> InterpolateOptions

Create new interpolate options with the given mode. Defaults to align_corners = true and no output size or scale factor.

pub fn with_align_corners(self, align_corners: bool) -> InterpolateOptions

Set align_corners.

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

Set the output size [height, width], clearing any scale factor.

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

Set the scale factor [scale_height, scale_width], clearing any output size.

Trait Implementations

impl Clone for InterpolateOptions

fn clone(&self) -> InterpolateOptions

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InterpolateOptions

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

Formats the value using the given formatter.

impl From<InterpolateOptions> for InterpolateOptionsIr

fn from(val: InterpolateOptions) -> InterpolateOptionsIr

Converts to this type from the input type.

impl From<InterpolateOptionsIr> for InterpolateOptions

fn from(val: InterpolateOptionsIr) -> InterpolateOptions

Converts to this type from the input type.