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

Set the i-th axis dimension.

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.

pub fn from_dims<T: AsRef<[TDim]> + Debug>(it: T) -> TractResult<ShapeInfo>

Trait Implementations

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 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 Debug for ShapeInfo

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

Formats the value using the given formatter.

impl TryFrom<()> for ShapeInfo

type Error = TractError

The type returned in the event of a conversion error.

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

Performs the conversion.

impl<'_> TryFrom<&'_ [TDim]> for ShapeInfo

type Error = TractError

The type returned in the event of a conversion error.

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

Performs the conversion.

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

type Error = TractError

The type returned in the event of a conversion error.

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

Performs the conversion.