def ce_checked

bendtorch-core@0.1.0.0 · src/autograd/loss.bend:395

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

395def ce_checked(logits: Core.Dynamic, +targets: List<&2, U32>, +rows: Nat, +columns: Nat, fits: Bool) ->
396  Core.Dynamic & Result<&2, &1, Error, Recorded>:
397  match fits:
398    case False{}:
399      (logits, Fail{TargetOutOfRange{}})
400    case True{}:
401      ce_targeted(logits, rows, columns, one_hot(targets, rows, columns), Shape.construct([U32.from_nat(rows), 1]))