def Mat.col_sums

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

import Base

290# sum of each column: Mat<n, m> -> Vec<m>
291def Mat.col_sums(-n: Nat, +m: Nat, a: Mat<n, m>) -> Vec<m>:
292  match a:
293    case Mat{x}:
294      Vec{rows.col_sums(m, x)}