def ce_shifted

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

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
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/reduction.bend as Reduction

363def ce_shifted(+rows: Nat, onehot: Core.Dynamic, +target: Shape.Shape, pair: (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>,
364  zero: Result<&2, &1, Core.Error, Core.Tensor<Reduction.scalar_shape()>>) -> Core.Dynamic & Result<&2, &1, Error, Recorded>:
365  (operands, result) = pair
366  (logits, maxes) = operands
367  match result zero:
368    case Fail{error} zero:
369      (logits, Fail{TensorError{error}})
370    case Done{shifted} Fail{error}:
371      (logits, Fail{TensorError{error}})
372    case Done{shifted} Done{tensor}:
373      (logits, ce_exps(rows, onehot, target,
374        Elementwise.binary(Elementwise.ExpLeft{}, shifted, Core.pack(Reduction.scalar_shape(), tensor))))