def Mat.col_sums

bend-ml-tensor-array@0.1.19.0 · main.bend:1379

import Base

1379def Mat.col_sums(+n: Nat, +m: Nat, a: Mat<n, m>) -> MColSum<n, m>:
1380  match a:
1381    case Mat{d}:
1382      colsum_fin(n, m, gemm(1n, m, n, 0, 1, U32.from_nat(m), 1, Array.new(F32, cap_depth(n), 1.0), d, Array.new(F32, cap_depth(m), 0.0)))