Struct AxesMapping

pub struct AxesMapping { /* private fields */ }

Implementations

impl AxesMapping

pub fn new( input_count: usize, output_count: usize, it: impl AsRef<[Axis]>, ) -> TractResult<AxesMapping>

pub fn for_numpy_matmul( rank: usize, transposing_a: bool, transposing_b: bool, transposing_c: bool, ) -> TractResult<AxesMapping>

pub fn disconnected( inputs: &[&TypedFact], outputs: &[&TypedFact], ) -> TractResult<AxesMapping>

pub fn disconnected_for_ranks( inputs: &[usize], outputs: &[usize], ) -> TractResult<AxesMapping>

pub fn natural( inputs: &[&TypedFact], outputs: &[&TypedFact], ) -> TractResult<AxesMapping>

pub fn natural_for_rank( inputs: usize, outputs: usize, rank: usize, ) -> TractResult<AxesMapping>

pub fn iter_all_axes(&self) -> impl Iterator<Item = &Axis>

pub fn iter_all_axes_mut(&mut self) -> impl Iterator<Item = &mut Axis>

pub fn input_count(&self) -> usize

pub fn output_count(&self) -> usize

pub fn axis_positions( &self, io: InOut, p: impl AxisPattern, ) -> TractResult<&[usize]>

pub fn rank(&self, io: InOut) -> usize

pub fn axis(&self, p: impl AxisPattern) -> TractResult<&Axis>

pub fn axes(&self, io: InOut) -> impl Iterator<Item = &Axis>

pub fn track_axis( &self, from: impl AxisPattern, to: InOut, ) -> TractResult<Option<usize>>

pub fn renaming( self, axis: impl AxisPattern, name: char, ) -> TractResult<AxesMapping>

pub fn linking( self, target: impl AxisPattern, axis: impl AxisPattern, ) -> TractResult<AxesMapping>

pub fn check(self) -> TractResult<AxesMapping>

pub fn available_label(&self) -> char

pub fn available_labels(&self) -> impl Iterator<Item = char> + '_

pub fn is_element_wise_unary(&self) -> bool

pub fn extract_sub_mapping( &self, inputs: &[usize], outputs: &[usize], ) -> TractResult<AxesMapping>

pub fn relabel(self) -> TractResult<AxesMapping>

pub fn remove_axis(&self, repr: char) -> TractResult<AxesMapping>

pub fn remove_axis_occurency( &self, slot: InOut, position: usize, ) -> TractResult<AxesMapping>

pub fn remove_slot(&self, slot: InOut) -> TractResult<AxesMapping>

pub fn with_extra_input(self, slot: usize) -> TractResult<AxesMapping>

pub fn with_extra_axis( self, repr: char, io: InOut, position: usize, ) -> TractResult<AxesMapping>

pub fn with_extra_axis_occurency( self, axis: impl AxisPattern, io: InOut, position: usize, ) -> TractResult<AxesMapping>

pub fn translate_to_axis_ops(&self) -> TractResult<Vec<AxisOp>>

pub fn from_strs( inputs: &[impl AsRef<str>], outputs: &[impl AsRef<str>], ) -> TractResult<AxesMapping>

pub fn to_strs(&self) -> (TVec<String>, TVec<String>)

pub fn change_axis_sink( &self, io: InOut, change: &AxisOp, ) -> TractResult<Option<AxesMapping>>

pub fn direct(&self, a: InOut, b: InOut) -> bool

pub fn same_layout<D: DimLike>( &self, a: InOut, b: InOut, shape_a: impl AsRef<[D]>, shape_b: impl AsRef<[D]>, ) -> bool

pub fn axis_ops_to_canonical(&self, io: InOut) -> TractResult<Vec<AxisOp>>

pub fn view_to_canonical<D>( &self, io: InOut, view: &mut ArrayViewD<'_, D>, ) -> TractResult<()>

pub fn view_to_canonical_mut<D>( &self, io: InOut, view: &mut ArrayViewMutD<'_, D>, ) -> TractResult<()>

pub fn compose(&self, other: &AxesMapping) -> TractResult<AxesMapping>

Trait Implementations

impl Clone for AxesMapping

fn clone(&self) -> AxesMapping

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for AxesMapping

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

Formats the value using the given formatter.

impl Display for AxesMapping

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

Formats the value using the given formatter.

impl FromStr for AxesMapping

type Err = Error

The associated error which can be returned from parsing.

fn from_str(s: &str) -> Result<Self, Self::Err>

Parses a string s to return a value of this type.

impl Hash for AxesMapping

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for AxesMapping

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

impl StructuralPartialEq for AxesMapping