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§
- Diag
Plan - Read-only diagonal extraction plan for one operand.
- Diag
Stage - Read-only metadata for one diagonal extraction stage.
- Gemm
Plan - Read-only GEMM decomposition plan for a pairwise contraction step.
- Pairwise
Step Plan - Read-only lowering data for one pairwise contraction step.
- Reduce
Plan - Read-only pre-reduction plan for axes unique to one operand.