Struct AxisUnionFind 

pub struct AxisUnionFind { /* private fields */ }

Union-Find data structure for grouping axis IDs.

Used to merge diagonal axes from Diag tensors so that they share the same representative ID when passed to einsum.

Implementations

impl AxisUnionFind

pub fn new() -> Self

Create a new empty union-find structure.

pub fn make_set(&mut self, id: DynId)

Add an ID to the structure (as its own set).

pub fn find(&mut self, id: DynId) -> DynId

Find the representative (root) of the set containing id. Uses path compression for efficiency.

pub fn union(&mut self, a: DynId, b: DynId)

Union the sets containing a and b. Uses union by rank for efficiency.

pub fn remap(&mut self, id: DynId) -> DynId

Remap an ID to its representative.

pub fn remap_ids(&mut self, ids: &[DynId]) -> Vec<DynId>

Remap a slice of IDs to their representatives.

Trait Implementations

impl Clone for AxisUnionFind

fn clone(&self) -> AxisUnionFind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AxisUnionFind

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

Formats the value using the given formatter.

impl Default for AxisUnionFind

fn default() -> Self

Returns the “default value” for a type.