Struct DiagPlan 

pub struct DiagPlan<'a> { /* private fields */ }

Read-only diagonal extraction plan for one operand.

Examples

use tenferro_einsum::{ContractionTree, Subscripts};

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

assert_eq!(diag.result_subs(), &[0]);

Implementations

impl<'a> DiagPlan<'a>

pub fn stages(self) -> impl ExactSizeIterator<Item = DiagStage<'a>> + 'a

Return the sequential diagonal extraction stages.

Examples

use tenferro_einsum::{ContractionTree, Subscripts};

let subs = Subscripts::new(&[&[0, 0], &[0]], &[0]);
let tree = ContractionTree::from_pairs(&subs, &[&[3, 3], &[3]], &[(0, 1)]).unwrap();
let diag = tree.step_plan(0).unwrap().lhs_diag().unwrap();
let mut stages = diag.stages();

assert_eq!(stages.next().unwrap().axis_pairs(), &[(0, 1)]);
assert!(stages.next().is_none());

pub fn result_subs(&self) -> &'a [u32]

Return final subscripts after all diagonal extraction stages.

Examples

use tenferro_einsum::{ContractionTree, Subscripts};

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

assert_eq!(diag.result_subs(), &[0]);

Trait Implementations

impl<'a> Clone for DiagPlan<'a>

fn clone(&self) -> DiagPlan<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for DiagPlan<'a>

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

Formats the value using the given formatter.

impl<'a> Copy for DiagPlan<'a>