Struct tract_core::model::ShapeFact

pub struct ShapeFact {
    pub stream_info: Option<StreamFact>,
    // 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<StreamFact>

Optional information for streaming tensors. None for regular tensors.

Methods

impl ShapeFact

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 set_dim(&mut self, i: usize, dim: TDim) -> TractResult<()>

Set the i-th axis dimension.

pub fn insert_axis(&mut self, axis: usize) -> TractResult<()>

pub fn remove_axis(&mut self, axis: usize) -> TractResult<()>

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 from_dims<T: AsRef<[TDim]> + Debug>(it: T) -> TractResult<ShapeFact>

Trait Implementations

impl Clone for ShapeFact

fn clone(&self) -> ShapeFact

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ShapeFact

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

Formats the value using the given formatter.

impl PartialEq<ShapeFact> for ShapeFact

fn eq(&self, other: &ShapeFact) -> 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<'_> TryFrom<&'_ [TDim]> for ShapeFact

type Error = TractError

The type returned in the event of a conversion error.

fn try_from(it: &[TDim]) -> TractResult<ShapeFact>

Performs the conversion.

impl<'_> TryFrom<&'_ [usize]> for ShapeFact

type Error = TractError

The type returned in the event of a conversion error.

fn try_from(it: &[usize]) -> TractResult<ShapeFact>

Performs the conversion.

impl TryFrom<()> for ShapeFact

type Error = TractError

The type returned in the event of a conversion error.

fn try_from(_it: ()) -> TractResult<ShapeFact>

Performs the conversion.