def ce_lse

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

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

333def ce_lse(+rows: Nat, onehot: Core.Dynamic, shifted: Core.Dynamic, result: Result<&2, &1, Core.Error, Core.Dynamic>) ->
334  Result<&2, &1, Error, Recorded>:
335  match result:
336    case Fail{error}:
337      Fail{TensorError{error}}
338    case Done{lse}:
339      ce_logged(rows, onehot, Elementwise.binary(Elementwise.Sub{}, shifted, lse))