def ce_loss2

bend-ml-autograd@0.1.1.0 · main.bend:282

import bend-ml-autograd@0.1.1.0/main.bend as BendMlAutograd

import Base
import bend-ml-tensor@0.1.0.0/main.bend as T

282def ce_loss2(-n: Nat, -c: Nat, +cc: Nat, +nf: F32, p: T.Mat<n, c>, labels: List<&2, Nat>) -> F32:
283  match p:
284    case T.Mat{rows}:
285      (0.0 - (rows.total(T.rows.zip_mul(log.rows(rows), onehot.rows(labels, cc)), 0.0) / nf : F32) : F32)

Uses (10)

Used by (1)

bend-ml-autograd@0.1.1.0/main:ce_loss