def Mat.matmul

bend-tensors@0.0.0.2 · bend_tensors.bend:1481

import Base

1481# a * b: Mat<rows, k> times Mat<k, cols> is Mat<rows, cols>
1482def Mat.matmul(+rows: Nat, +k: Nat, +cols: Nat, a: Mat<rows, k>, b: Mat<k, cols>) -> Mat<rows, cols>:
1483  MatS{x} = a
1484  MatS{y} = b
1485  MatS{MatT.mask(VecT.depth(rows), VecT.depth(cols), MatT.matmul(VecT.depth(rows), VecT.depth(k), VecT.depth(cols), x, y), 0n, 0n, rows, cols)}