Struct IoSpec 

pub struct IoSpec {
    pub name: String,
    pub dtype: DataType,
    pub shape: Vec<Option<usize>>,
}

A model input or output: its name, element type and per-item shape (without the batch dimension; None where the graph leaves a dimension open)

Fields

name: String

dtype: DataType

shape: Vec<Option<usize>>

Implementations

impl IoSpec

pub fn new(name: &str, dtype: DataType, shape: &[Option<usize>]) -> Self

A tensor of dtype with per-item shape (None for a dimension each request fixes)

pub fn fp32(name: &str, shape: &[usize]) -> Self

An f32 tensor with a fixed per-item shape

pub fn static_shape(&self) -> Option<Vec<usize>>

The per-item shape if every dimension is fixed

pub fn v2_shape(&self) -> Vec<i64>

For metadata: the full shape with -1 for the batch and for open dimensions

Trait Implementations

impl Clone for IoSpec

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for IoSpec

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

Formats the value using the given formatter.

impl From<&IoSpec> for TensorInfo

fn from(spec: &IoSpec) -> Self

Converts to this type from the input type.

impl PartialEq for IoSpec

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for IoSpec