def ce_checked

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

import Base

1455def ce_checked(ok: Bool, +n: Nat, +c: Nat, z: Mat<n, c>, labels: List<&2, Nat>) -> Maybe<&1, MCE<n, c>>:
1456  match ok:
1457    case True{}:
1458      Some{Mat.softmax_ce(n, c, z, labels)}
1459    case False{}:
1460      None{}