def one_hot_pair

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

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

286def one_hot_pair(targets: Result<&2, &1, Core.Error, Core.Dynamic>, columns: Result<&2, &1, Core.Error, Core.Dynamic>) ->
287  Result<&2, &1, Core.Error, Core.Dynamic>:
288  match targets columns:
289    case Fail{error} columns:
290      Fail{error}
291    case Done{targets} Fail{error}:
292      Fail{error}
293    case Done{targets} Done{columns}:
294      quotient(Elementwise.binary(Elementwise.Matches{}, targets, columns))