~/jendhub

def ce_leaf

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

import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
import Base

367# rows i0 .. i0+cnt-1: their gradients in the logits, their loss terms in t
368def ce_leaf(+e: SE, +i0: U32, +cnt: Nat, g: B3<F32, U32, F32>) -> B3<F32, U32, F32>:
369  match e:
370    case SE{+mm1, +mu, +nf}:
371      ce_loop(cnt, mm1, mu, nf, i0, g)

Uses (6)

Used by (1)