Skip to main content

matmul

Function matmul 

Source
pub fn matmul(a: &TracedTensor, b: &TracedTensor) -> TracedTensor
Expand description

Matrix multiplication helper for rank-2 traced tensors.

This contracts the last dimension of a with the first dimension of b.

§Examples

let c = tenferro::matmul(&a, &b);