pub fn _contract<T: Zero + ComplexFloat, La: Layout, Lb: Layout>(
bd: impl MatMul<T>,
a: &Slice<T, DynRank, La>,
b: &Slice<T, DynRank, Lb>,
axes: Axes,
alpha: T,
) -> Tensor<T, DynRank>Expand description
Helper for implementing contraction through matrix multiplication