Skip to main content

contract_native_tensor

Function contract_native_tensor 

Source
pub fn contract_native_tensor(
    lhs: &Tensor,
    axes_a: &[usize],
    rhs: &Tensor,
    axes_b: &[usize],
) -> Result<Tensor>
Expand description

Contract two native tensors along matching axes.

ยงErrors

Returns an error when axes_a/axes_b are out of bounds or have differing lengths, when the contracted dimensions do not match, or when the backend einsum fails.