def Mat.matmul.sq

bend-tensors@0.0.0.2 · bend_tensors.bend:1487

import Base

1487# a * b for two Mat<n, n>: the one-depth recursion, 3x faster per node
1488def Mat.matmul.sq(+n: Nat, a: Mat<n, n>, b: Mat<n, n>) -> Mat<n, n>:
1489  MatS{x} = a
1490  MatS{y} = b
1491  MatS{MatT.mask(VecT.depth(n), VecT.depth(n), MatT.matmul.sq(VecT.depth(n), x, y), 0n, 0n, n, n)}