Struct tract_core::analyser::types::ShapeFact

pub struct ShapeFact { /* fields omitted */ }

Partial information about a shape.

A basic example of a shape fact is shapefact![1, 2], which corresponds to the shape [1, 2] in Arc. We can use _ in facts to denote unknown dimensions (e.g. shapefact![1, 2, _] corresponds to any shape [1, 2, k] with k a non-negative integer). We can also use .. at the end of a fact to only specify its first dimensions, so shapefact![1, 2; ..] matches any shape that starts with [1, 2] (e.g. [1, 2, i] or [1, 2, i, j]), while shapefact![..] matches any shape.

Methods

impl ShapeFact

pub fn open(dims: TVec<DimFact>) -> ShapeFact

Constructs an open shape fact.

pub fn is_open(&self) -> bool

pub fn closed(dims: TVec<DimFact>) -> ShapeFact

Constructs a closed shape fact.

pub fn rank(&self) -> IntFact

pub fn dims(&self) -> impl Iterator<Item = DimFact>

pub fn stream_info(&self) -> TractResult<Option<StreamInfo>>

pub fn as_concrete_finite(&self) -> TractResult<Option<TVec<usize>>>

Trait Implementations

impl Fact for ShapeFact

type Concrete = TVec<TDim>

fn concretize(self: &ShapeFact) -> Option<TVec<TDim>>

Tries to transform the fact into a Vec<usize>, or returns None.

fn unify(&self, other: &Self) -> TractResult<Self>

Tries to unify the fact with another fact of the same type.

fn is_concrete(&self) -> bool

Returns whether the value is fully determined.

impl Output for ShapeFact

fn into_wrapped(source: Self) -> Wrapped

Wraps the fact in the Wrapped type.

fn from_wrapped(wrapped: Wrapped) -> TractResult<ShapeFact>

Retrieves the fact from the Wrapped type. Panics if wrapped doesn't have the right constructor.

fn wrap(self) -> Wrapped

Wraps self in the Wrapped type.

impl IntoExp<ShapeFact> for ShapeFact

fn bex(self) -> Exp<ShapeFact>

Converts the value to an Expression.

impl IntoExp<ShapeFact> for ShapeProxy

fn bex(self) -> Exp<ShapeFact>

Converts the value to an Expression.

impl<'a> IntoExp<ShapeFact> for &'a ShapeProxy

fn bex(self) -> Exp<ShapeFact>

Converts the value to an Expression.

impl IntoExp<ShapeFact> for TVec<TDim>

fn bex(self) -> Exp<ShapeFact>

Converts the value to an Expression.

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 ==.

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

This method tests for !=.

impl Default for ShapeFact

fn default() -> ShapeFact

Returns the most general shape fact possible.

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<D: ToDim, I: IntoIterator<Item = D>> From<I> for ShapeFact

fn from(it: I) -> ShapeFact

Performs the conversion.

impl Debug for ShapeFact

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

Formats the value using the given formatter.

impl FromIterator<TDim> for ShapeFact

fn from_iter<I: IntoIterator<Item = TDim>>(iter: I) -> ShapeFact

Converts an iterator over usize into a closed shape.

impl FromIterator<usize> for ShapeFact

fn from_iter<I: IntoIterator<Item = usize>>(iter: I) -> ShapeFact

Converts an iterator over usize into a closed shape.