def dims_summed

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1380

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

1380def dims_summed(node: Node, entries: List<&1, Entry>, pair: Core.Dynamic & Result<&2, &1, SumDims.Error, Core.Dynamic>) ->
1381  Tape & Result<&2, &2, Error, Nat>:
1382  (input, result) = pair
1383  match result:
1384    case Fail{error}:
1385      (Tape{entries}, Fail{dims_error(error)})
1386    case Done{output}:
1387      push(node, output, Unsaved{}, entries)