Struct ndarray_einsum_beta::contractors::pair_contractors::StackedTensordotGeneral

pub struct StackedTensordotGeneral {
    lhs_permutation: Permutation,
    rhs_permutation: Permutation,
    lhs_output_shape: Vec<usize>,
    rhs_output_shape: Vec<usize>,
    intermediate_shape: Vec<usize>,
    tensordot_fixed_position: TensordotFixedPosition,
    output_shape: Vec<usize>,
    output_permutation: Permutation,
}

Repeatedly computes the tensor dot of subviews of the two tensors, iterating over indices which appear in the LHS, RHS, and output.

The indices appearing in all three places are referred to here as the “stack” indices. For example, in the contraction ijk,ikl->ijl, i would be the (only) “stack” index. This contraction is an instance of batch matrix multiplication. The LHS and RHS are both 3-D tensors and the ith (2-D) subview of the output is the matrix product of the ith subview of the LHS matrix-multiplied by the ith subview of the RHS.

This is the most general contraction and in theory could handle all pairwise contractions, but is less performant than special-casing when there are no “stack” indices. It is also currently the only case that requires .outer_iter_mut() (which might make parallelizing operations more difficult).

Fields

lhs_permutation: Permutation

rhs_permutation: Permutation

lhs_output_shape: Vec<usize>

rhs_output_shape: Vec<usize>

intermediate_shape: Vec<usize>

tensordot_fixed_position: TensordotFixedPosition

output_shape: Vec<usize>

output_permutation: Permutation

Implementations

impl StackedTensordotGeneral

pub fn new(sc: &SizedContraction) -> Self

Trait Implementations

impl Clone for StackedTensordotGeneral

fn clone(&self) -> StackedTensordotGeneral

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for StackedTensordotGeneral

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

Formats the value using the given formatter.

impl<A> PairContractor<A> for StackedTensordotGeneral

fn contract_pair<'a, 'b, 'c, 'd>(
    &self,
    lhs: &'b ArrayViewD<'a, A>,
    rhs: &'d ArrayViewD<'c, A>
) -> ArrayD<A> where
    'a: 'b,
    'c: 'd,
    A: Clone + LinalgScalar, 

fn contract_and_assign_pair<'a, 'b, 'c, 'd, 'e, 'f>(
    &self,
    lhs: &'b ArrayViewD<'a, A>,
    rhs: &'d ArrayViewD<'c, A>,
    out: &'f mut ArrayViewMutD<'e, A>
) where
    'a: 'b,
    'c: 'd,
    'e: 'f,
    A: Clone + LinalgScalar,