Struct Input

pub struct Input {
    pub name: String,
    pub input_type: TensorElementDataType,
    pub dimensions: Vec<Option<u32>>,
}

Information about an ONNX’s input as stored in loaded file

Fields

name: String

Name of the input layer

input_type: TensorElementDataType

Type of the input layer’s elements

dimensions: Vec<Option<u32>>

Shape of the input layer

C API uses a i64 for the dimensions. We use an unsigned of the same range of the positive values.

Implementations

impl Input

pub fn dimensions(&self) -> impl Iterator<Item = Option<usize>> + '_

Return an iterator over the shape elements of the input layer

Note: The member Input::dimensions stores u32 (since ONNX uses i64 but which cannot be negative) so the iterator converts to usize.

Trait Implementations

impl Debug for Input

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

Formats the value using the given formatter.