def ce_logged
bendtorch-core@0.1.0.0 · src/autograd/loss.bend:323
import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Loss
import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
323def ce_logged(+rows: Nat, onehot: Core.Dynamic, pair: (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>) ->
324 Result<&2, &1, Error, Recorded>:
325 (operands, result) = pair
326 (shifted, lse) = operands
327 match result:
328 case Fail{error}:
329 Fail{TensorError{error}}
330 case Done{lp}:
331 ce_picked(rows, Elementwise.binary(Elementwise.PickNegated{}, lp, onehot))