def count_rows3

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

import Base

514def count_rows3(+y: Nat, n: Nat, al: AL) -> CR:
515  match al:
516    case AL{c, l}:
517      CR{c, Nat.add(n, one(Nat.is_eq(argmax(l), y)))}