Struct Output

pub struct Output {
    pub name: String,
    pub output_type: TensorElementDataType,
    pub dimensions: Vec<Option<u32>>,
}

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

Fields

name: String

Name of the output layer

output_type: TensorElementDataType

Type of the output layer’s elements

dimensions: Vec<Option<u32>>

Shape of the output layer

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

Implementations

impl Output

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

Return an iterator over the shape elements of the output layer

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

Trait Implementations

impl Debug for Output

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

Formats the value using the given formatter.