Struct EinsumNode 

pub struct EinsumNode {
    pub op: OpType,
    pub inputs: Vec<usize>,
    pub outputs: Vec<usize>,
    pub metadata: Option<Metadata>,
}

Fields

op: OpType

inputs: Vec<usize>

outputs: Vec<usize>

Tensor indices that this node produces/writes to. Most operations produce a single tensor, but some may produce multiple.

metadata: Option<Metadata>

Optional metadata for debugging and provenance tracking

Implementations

impl EinsumNode

pub fn new( spec: impl Into<String>, inputs: Vec<usize>, outputs: Vec<usize>, ) -> Self

pub fn einsum( spec: impl Into<String>, inputs: Vec<usize>, outputs: Vec<usize>, ) -> Self

pub fn elem_unary(op: impl Into<String>, input: usize, output: usize) -> Self

pub fn elem_binary( op: impl Into<String>, left: usize, right: usize, output: usize, ) -> Self

pub fn reduce( op: impl Into<String>, axes: Vec<usize>, input: usize, output: usize, ) -> Self

pub fn with_single_output( spec: impl Into<String>, inputs: Vec<usize>, output_idx: usize, ) -> Self

Creates a node with automatic output tracking. The output tensor index should be provided by the caller after calling add_tensor(). This is a convenience method for the common case of single-output operations.

pub fn validate(&self, num_tensors: usize) -> Result<(), IrError>

pub fn primary_output(&self) -> Option<usize>

Get the primary output tensor index (first output). Most operations produce a single tensor.

pub fn produces(&self, tensor_idx: usize) -> bool

Check if this node produces a specific tensor.

pub fn parse_einsum_spec(&self) -> Result<Option<EinsumSpec>, IrError>

Parse and validate the einsum spec if this is an Einsum operation.

pub fn operation_description(&self) -> String

Get a human-readable description of this node’s operation.

pub fn with_metadata(self, metadata: Metadata) -> Self

Attach metadata to this node.

pub fn get_metadata(&self) -> Option<&Metadata>

Get the metadata if present.

pub fn set_metadata(&mut self, metadata: Metadata)

Set the metadata for this node.

Trait Implementations

impl Clone for EinsumNode

fn clone(&self) -> EinsumNode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EinsumNode

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for EinsumNode

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for EinsumNode

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

Formats the value using the given formatter.

impl PartialEq for EinsumNode

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

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for EinsumNode