def ce_loss

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

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

287# mean cross-entropy loss: -(1/n) Σ log p[i][label_i]
288def ce_loss(+n: Nat, +c: Nat, logits: T.Mat<n, c>, labels: List<&2, Nat>) -> F32:
289  ce_loss2(n, c, c, F32.from_nat(n), T.Mat.softmax(n, c, logits), labels)