Struct TensordotFixedPosition 

pub struct TensordotFixedPosition {
    len_uncontracted_lhs: usize,
    len_uncontracted_rhs: usize,
    len_contracted_axes: usize,
    output_shape: Vec<usize>,
}

Performs tensor dot product for two tensors where no permutation needs to be performed, e.g. ijk,jkl->il or ijk,klm->ijlm.

The axes to be contracted must be the last axes of the LHS tensor and the first axes of the RHS tensor, and the axis order for the output tensor must be all the uncontracted axes of the LHS tensor followed by all the uncontracted axes of the RHS tensor, in the orders those originally appear in the LHS and RHS tensors.

The contraction is performed by reshaping the LHS into a matrix (2-D tensor) of shape [len_uncontracted_lhs, len_contracted_axes], reshaping the RHS into shape [len_contracted_axes, len_contracted_rhs], matrix-multiplying the two reshaped tensor, and then reshaping the result into […self.output_shape].

Fields

len_uncontracted_lhs: usize

The product of the lengths of all the uncontracted axes in the LHS (or 1 if all of the LHS axes are contracted)

len_uncontracted_rhs: usize

The product of the lengths of all the uncontracted axes in the RHS (or 1 if all of the RHS axes are contracted)

len_contracted_axes: usize

The product of the lengths of all the contracted axes (or 1 if no axes are contracted, i.e. the outer product is computed)

output_shape: Vec<usize>

The shape that the tensor dot product will be recast to

Implementations

impl TensordotFixedPosition

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

pub fn from_shapes_and_number_of_contracted_axes( lhs_shape: &[usize], rhs_shape: &[usize], num_contracted_axes: usize, ) -> Self

Compute the uncontracted and contracted axis lengths and the output shape based on the input shapes and how many axes should be contracted from each tensor.

TODO: The assert_eq! here could be tightened up by verifying that the last num_contracted_axes of the LHS match the first num_contracted_axes of the RHS axis-by-axis (as opposed to only checking the product as is done here.)

Trait Implementations

impl Clone for TensordotFixedPosition

fn clone(&self) -> TensordotFixedPosition

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensordotFixedPosition

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

Formats the value using the given formatter.

impl<A> PairContractor<A> for TensordotFixedPosition

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,