Struct InputShape 

pub struct InputShape {
    pub batch: Dim,
    pub channels: Dim,
    pub height: Dim,
    pub width: Dim,
}

Input shape specification for an ONNX model. Represents [batch, channels, height, width] with support for dynamic dimensions.

Fields

batch: Dim

Batch dimension

channels: Dim

Channel dimension

height: Dim

Height dimension

width: Dim

Width dimension

Implementations

impl InputShape

pub fn new(batch: Dim, channels: Dim, height: Dim, width: Dim) -> InputShape

Creates a new input shape from individual dimensions.

pub fn from_array(shape: [i64; 4]) -> InputShape

Creates an input shape from an array of i64 values. Values <= 0 are treated as dynamic dimensions.

pub fn fixed(batch: i64, channels: i64, height: i64, width: i64) -> InputShape

Creates a fully fixed input shape.

pub fn dynamic_batch(channels: i64, height: i64, width: i64) -> InputShape

Creates a shape with dynamic batch and fixed spatial dimensions. Common pattern: [-1, 3, H, W]

pub fn dynamic_spatial(batch: i64, channels: i64) -> InputShape

Creates a shape with fixed batch/channels and dynamic spatial dimensions. Common pattern: [1, 3, -1, -1]

pub fn fully_dynamic() -> InputShape

Creates a fully dynamic shape: [-1, -1, -1, -1]

pub fn spatial_size_or(&self, default_h: u32, default_w: u32) -> (u32, u32)

Returns the spatial dimensions as (height, width) with defaults for dynamic dims.

pub fn has_fixed_spatial(&self) -> bool

Returns whether the spatial dimensions (height, width) are fixed.

pub fn has_dynamic(&self) -> bool

Returns whether any dimension is dynamic.

pub fn to_array(&self) -> [i64; 4]

Converts to array representation where dynamic = -1.

pub fn from_onnx_dims(dims: &[i64]) -> Option<InputShape>

Parses input shape from ONNX model input dimensions.

Handles various dimension representations:

• Positive values: fixed dimensions
• Negative values or 0: dynamic dimensions

Arguments

• dims - Dimensions from ONNX model (typically 4 elements: [batch, channels, height, width])

Returns

• Some(InputShape) if dims has exactly 4 elements
• None if dims has wrong number of elements

Trait Implementations

impl Clone for InputShape

fn clone(&self) -> InputShape

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InputShape

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

Formats the value using the given formatter.

impl Default for InputShape

fn default() -> InputShape

Returns the “default value” for a type.

impl Display for InputShape

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

Formats the value using the given formatter.

impl PartialEq for InputShape

fn eq(&self, other: &InputShape) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for InputShape

impl StructuralPartialEq for InputShape