Struct TensorSpec 

pub struct TensorSpec {
    pub name: String,
    pub dtype: DataType,
    pub shape: Vec<Dimension>,
}

One named model input or output contract.

Fields

name: String

ONNX/model-visible name.

dtype: DataType

Required scalar/vector dtype.

shape: Vec<Dimension>

Ordered fixed or dynamic dimensions.

Implementations

impl TensorSpec

pub fn new( name: impl Into<String>, dtype: DataType, shape: Vec<Dimension>, ) -> Self

Creates a named tensor specification.

pub fn validate(&self, descriptor: &TensorDescriptor) -> AiResult<()>

Validates a concrete tensor descriptor against this specification.

Trait Implementations

impl Clone for TensorSpec

fn clone(&self) -> TensorSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorSpec

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

Formats the value using the given formatter.

impl PartialEq for TensorSpec

fn eq(&self, other: &TensorSpec) -> 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 TensorSpec

impl StructuralPartialEq for TensorSpec