def ce_dimensioned

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

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

403def ce_dimensioned(logits: Core.Dynamic, +targets: List<&2, U32>, +dimensions: List<&2, U32>) ->
404  Core.Dynamic & Result<&2, &1, Error, Recorded>:
405  match dimensions:
406    case [+rows, +columns]:
407      ce_checked(logits, targets, U32.to_nat(rows), U32.to_nat(columns), targets_fit(targets, U32.to_nat(rows), U32.to_nat(columns)))
408    case _:
409      (logits, Fail{NotLogits{}})