def ce_targeted

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

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

385def ce_targeted(logits: Core.Dynamic, +rows: Nat, +columns: Nat, onehot: Result<&2, &1, Core.Error, Core.Dynamic>,
386  result: Shape.ConstructResult<[U32.from_nat(rows), 1]>) -> Core.Dynamic & Result<&2, &1, Error, Recorded>:
387  match onehot result:
388    case Fail{error} result:
389      (logits, Fail{TensorError{error}})
390    case Done{onehot} Shape.SError{error}:
391      (logits, Fail{TensorError{Core.ShapeError{error}}})
392    case Done{onehot} Shape.SBuilt{+target, preserved}:
393      ce_maxed(rows, onehot, target, row_maxes(logits, target, columns))