def Mat.col_sums_acc

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

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray

import Base

1387def Mat.col_sums_acc(+n: Nat, +m: Nat, a: Mat<n, m>, s: Mat<1n, m>) -> MColSum<n, m>:
1388  match a s:
1389    case Mat{d} Mat{z}:
1390      colsum_fin(n, m, gemm_ac(True{}, 1n, m, n, 0, 1, U32.from_nat(m), 1, Array.new(F32, cap_depth(n), 1.0), d, z))

Uses (10)

Used by (2)