Struct BroadcastProductGeneral 

pub struct BroadcastProductGeneral {
    lhs_permutation: Permutation,
    rhs_permutation: Permutation,
    lhs_insertions: Vec<usize>,
    rhs_insertions: Vec<usize>,
    output_sizes: Vec<usize>,
    hadamard_product: HadamardProduct,
}

Permutes the axes of the LHS and RHS tensor, broadcasts into the output shape, and then computes the element-wise product of the two broadcast tensors.

Currently unused due to (limited) unfavorable benchmarking results compared to StackedTensordotGeneral. An example of a contraction that could theoretically be performed by this contraction is ij,jk->ijk: the LHS and RHS are both broadcast into output shape (|i|, |j|, |k|) and then multiplied elementwise.

However, the limited benchmarking performed so far favored iterating along the j axis and computing the outer products i,k->ik for each subview of the tensors.

Fields

lhs_permutation: Permutation

rhs_permutation: Permutation

lhs_insertions: Vec<usize>

rhs_insertions: Vec<usize>

output_sizes: Vec<usize>

hadamard_product: HadamardProduct

Implementations

impl BroadcastProductGeneral

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

Trait Implementations

impl Clone for BroadcastProductGeneral

fn clone(&self) -> BroadcastProductGeneral

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BroadcastProductGeneral

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

Formats the value using the given formatter.

impl<A> PairContractor<A> for BroadcastProductGeneral

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

where A: Clone + LinalgScalar, 'a: 'b, 'c: 'd,

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: Clone + LinalgScalar, 'a: 'b, 'c: 'd, 'e: 'f,