def Mat.softmax_ce

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

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

import Base

1403def Mat.softmax_ce(+n: Nat, +c: Nat, z: Mat<n, c>, labels: List<&2, Nat>) -> MCE<n, c>:
1404  match z:
1405    case Mat{d}:
1406      ce_fin(n, c, F32.from_nat(n), cea_rows(labels, Nat.sub(c, 1n), U32.from_nat(c), F32.from_nat(n), 0, L2{0.0, d}))

Uses (10)

Used by (2)