Struct PadDescriptor 

pub struct PadDescriptor<const N: usize> {
    pub mode: PadMode,
    pub input_shape: [i32; N],
    pub pad_low: [i32; N],
    pub pad_high: [i32; N],
    pub value: f32,
    pub element: ElementKind,
}

Descriptor for a constant-pad op.

input_shape is the shape of the input tensor. pad_low[d] and pad_high[d] are the pad amounts on each side of axis d. Output shape is input_shape[d] + pad_low[d] + pad_high[d] per axis. value is the constant used in the pad region (for Constant mode). element must match T::KIND at select time.

Fields

mode: PadMode

Padding mode — one of PadMode::Constant / PadMode::Reflect / PadMode::Replicate / PadMode::Circular. All four are wired for every supported dtype.

input_shape: [i32; N]

Input tensor shape.

pad_low: [i32; N]

Pad amount on the low side of each axis. Non-negative.

pad_high: [i32; N]

Pad amount on the high side of each axis. Non-negative.

value: f32

Constant value used in the pad region for Constant mode.

element: ElementKind

Element type of input and output.

Implementations

impl<const N: usize> PadDescriptor<N>

pub fn output_shape(&self) -> [i32; N]

Compute the output shape from input shape + pad amounts.

Trait Implementations

impl<const N: usize> Clone for PadDescriptor<N>

fn clone(&self) -> PadDescriptor<N>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<const N: usize> Copy for PadDescriptor<N>

impl<const N: usize> Debug for PadDescriptor<N>

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

Formats the value using the given formatter.