def softmax_ce2

bend-ml-tensor-fork@0.1.6.0 · main.bend:414

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

414def softmax_ce2(seq: Bool, +n: Nat, +c: Nat, +d: Nat, z: TA.Mat<n, c>, labels: List<&2, Nat>) -> TA.MCE<n, c>:
415  match seq z:
416    case True{} z:
417      TA.Mat.softmax_ce(n, c, z, labels)
418    case False{} TA.Mat{x}:
419      ce_par3(n, c, F32.from_nat(n), Nat.sub(c, 1n), U32.from_nat(c), d, x, lab_take(n, labels, Nat.sub(c, 1n), c, 0, 0n, Array.new(U32, depth(n), 0)))