Enum IndexExpr 

pub enum IndexExpr {
    Var(String),
    Const(i64),
    Add(Box<IndexExpr>, Box<IndexExpr>),
    Sub(Box<IndexExpr>, Box<IndexExpr>),
    Mul(Box<IndexExpr>, Box<IndexExpr>),
    Div(Box<IndexExpr>, Box<IndexExpr>),
    Min(Box<IndexExpr>, Box<IndexExpr>),
    Max(Box<IndexExpr>, Box<IndexExpr>),
}

Index expression for dimension calculations.

Index expressions represent compile-time or runtime values used in type indices, particularly for tensor dimensions.

Variants

Var(String)

Variable index (e.g., n, m)

Const(i64)

Constant index value

Add(Box<IndexExpr>, Box<IndexExpr>)

Addition: e1 + e2

Sub(Box<IndexExpr>, Box<IndexExpr>)

Subtraction: e1 - e2

Mul(Box<IndexExpr>, Box<IndexExpr>)

Multiplication: e1 * e2

Div(Box<IndexExpr>, Box<IndexExpr>)

Division: e1 / e2

Min(Box<IndexExpr>, Box<IndexExpr>)

Minimum: min(e1, e2)

Max(Box<IndexExpr>, Box<IndexExpr>)

Maximum: max(e1, e2)

Implementations

impl IndexExpr

pub fn var(name: impl Into<String>) -> Self

Create a variable index expression

pub fn constant(value: i64) -> Self

Create a constant index expression

pub fn add(left: IndexExpr, right: IndexExpr) -> Self

Addition

pub fn sub(left: IndexExpr, right: IndexExpr) -> Self

Subtraction

pub fn mul(left: IndexExpr, right: IndexExpr) -> Self

Multiplication

pub fn div(left: IndexExpr, right: IndexExpr) -> Self

Division

pub fn min(left: IndexExpr, right: IndexExpr) -> Self

Minimum

pub fn max(left: IndexExpr, right: IndexExpr) -> Self

Maximum

pub fn free_vars(&self) -> HashSet<String>

Get all free variables in this expression

pub fn substitute(&self, subst: &HashMap<String, IndexExpr>) -> IndexExpr

Substitute variables with index expressions

pub fn simplify(&self) -> IndexExpr

Simplify the index expression

pub fn try_eval(&self) -> Option<i64>

Try to evaluate to a constant value

Trait Implementations

impl Clone for IndexExpr

fn clone(&self) -> IndexExpr

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for IndexExpr

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for IndexExpr

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for IndexExpr

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

Formats the value using the given formatter.

impl Hash for IndexExpr

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 IndexExpr

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

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Eq for IndexExpr

impl StructuralPartialEq for IndexExpr