~/jendhub

def ce_row

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

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

384def ce_row(al: AL, +label: Nat, +nf: F32, +idx: U32, loss: F32) -> L2:
385  match al:
386    case AL{c, l}:
387      ce_row2(softmax(l), label, nf, idx, loss, c)

Uses (7)

Used by (1)