Struct tract_core::model::ShapeInfo

pub struct ShapeInfo {
    pub stream_info: Option<StreamInfo>,
    // some fields omitted
}

Fully determined dimension of a tensor.

Tensors in tract can have one streaming dimension. TDim generalize the regular tensor dimensions (usize) to arithmetic expressions of S, the (sometimes hypothetical) tensor length on the streaming axis.

Fields

stream_info: Option<StreamInfo>

Optional information for streaming tensors. None for regular tensors.

Methods

impl ShapeInfo

pub fn rank(&self) -> usize

Rank of the tensor.

pub fn dim(&self, i: usize) -> TDim

Extended dimension of the i-th axis.

The TDim will wrap a plain integer for regular (non-streaming) tensors.

pub fn as_finite(&self) -> Option<&[usize]>

Shape of the tensor, unless it is streaming.

pub fn iter<'a>(&'a self) -> impl Iterator<Item = TDim> + 'a

Iterator over dimension of the shape.

pub fn to_tvec(&self) -> TVec<TDim>

Convert the shape to an array of extended dimensions.

pub fn to_shape_fact(&self) -> ShapeFact

Convert the shape to a fully determined shape fact.

Trait Implementations

impl PartialEq<ShapeInfo> for ShapeInfo

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

This method tests for self and other values to be equal, and is used by ==.

#[must_use]

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

This method tests for !=.

impl Clone for ShapeInfo

fn clone(&self) -> ShapeInfo

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: AsRef<[usize]>> From<T> for ShapeInfo

fn from(it: T) -> ShapeInfo

Performs the conversion.

impl Debug for ShapeInfo

fn fmt(&self, fmt: &mut Formatter) -> Result

Formats the value using the given formatter.