Skip to main content

contract

Function contract 

Source
pub fn contract<R: Runtime>(
    a: &RudaTensor<R>,
    modes_a: &[Mode],
    b: &RudaTensor<R>,
    modes_b: &[Mode],
    output_modes: &[Mode],
) -> Result<RudaTensor<R>>
Expand description

Sum the product of A and B over modes absent from the output.