def Mat.matmul_t

bend-ml-tensor@0.1.2.0 · main.bend:276

import Base

276# (n x k) · (m x k)ᵀ = (n x m): the second operand arrives already TRANSPOSED (stored as
277# m rows of k numbers). It avoids transposing large weights on every call; the type
278# still requires the same inner dimension k.
279def Mat.matmul_t(-n: Nat, -k: Nat, -m: Nat, a: Mat<n, k>, bt: Mat<m, k>) -> Mat<n, m>:
280  match a bt:
281    case Mat{x} Mat{y}:
282      Mat{matmul.rows(x, y)}