def softmax_ce

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

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

421# mean cross-entropy and its gradient, TA.Mat.softmax_ce's arguments and bits, with the rows in up to 2^par blocks
422# (par = 0: tensor-array's own loop)
423def softmax_ce(+n: Nat, +c: Nat, +par: Nat, z: TA.Mat<n, c>, labels: List<&2, Nat>) -> TA.MCE<n, c>:
424  softmax_ce2(Bool.or(Nat.is_eq(par, 0n), Nat.is_eq(c, 0n)), n, c, par, z, labels)