Struct EinsumSubscripts 

pub struct EinsumSubscripts {
    pub inputs: Vec<Vec<u32>>,
    pub output: Vec<u32>,
}

Canonical N-ary einsum subscripts using integer labels.

String notation is a user-facing convenience. Runtime integration layers can carry this representation in extension payloads so execution, shape inference, and AD do not need to parse strings.

Fields

inputs: Vec<Vec<u32>>

Index labels for each input tensor.

output: Vec<u32>

Index labels for the output tensor.

Implementations

impl EinsumSubscripts

pub fn new(inputs: &[&[u32]], output: &[u32]) -> Self

Create subscripts from integer label arrays.

Examples

use tenferro_einsum::EinsumSubscripts;

let subscripts = EinsumSubscripts::new(&[&[0, 1], &[1, 2]], &[0, 2]);

assert_eq!(subscripts.inputs, vec![vec![0, 1], vec![1, 2]]);
assert_eq!(subscripts.output, vec![0, 2]);

pub fn input_count(&self) -> usize

Number of input operands described by this specification.

Examples

use tenferro_einsum::EinsumSubscripts;

let subscripts = EinsumSubscripts::new(&[&[0], &[0]], &[]);

assert_eq!(subscripts.input_count(), 2);

Trait Implementations

impl Clone for EinsumSubscripts

fn clone(&self) -> EinsumSubscripts

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EinsumSubscripts

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

Formats the value using the given formatter.

impl From<&EinsumSubscripts> for Subscripts

fn from(subscripts: &EinsumSubscripts) -> Self

Converts to this type from the input type.

impl From<&Subscripts> for EinsumSubscripts

fn from(subscripts: &Subscripts) -> Self

Converts to this type from the input type.

impl From<EinsumSubscripts> for Subscripts

fn from(subscripts: EinsumSubscripts) -> Self

Converts to this type from the input type.

impl From<Subscripts> for EinsumSubscripts

fn from(subscripts: Subscripts) -> Self

Converts to this type from the input type.

impl Hash for EinsumSubscripts

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 EinsumSubscripts

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

impl StructuralPartialEq for EinsumSubscripts