Skip to main content

Module lowering

Module lowering 

Source
Expand description

Read-only einsum lowering plans.

These wrappers expose shape and mode metadata computed by the ordinary einsum planner without exposing graph-building or arithmetic execution.

§Examples

use tenferro_einsum::{ContractionTree, Subscripts};

let subs = Subscripts::new(&[&[0, 1], &[1, 2]], &[0, 2]);
let tree = ContractionTree::from_pairs(&subs, &[&[2, 3], &[3, 4]], &[(0, 1)]).unwrap();
let gemm = tree.step_plan(0).unwrap().gemm();

assert_eq!(gemm.contracted_modes(), &[1]);

Structs§

DiagPlan
Read-only diagonal extraction plan for one operand.
DiagStage
Read-only metadata for one diagonal extraction stage.
GemmPlan
Read-only GEMM decomposition plan for a pairwise contraction step.
PairwiseStepPlan
Read-only lowering data for one pairwise contraction step.
ReducePlan
Read-only pre-reduction plan for axes unique to one operand.