def summed

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

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

1347def summed(+a: Nat, entries: List<&1, Entry>, +shape: Shape.Shape, result: Result<&2, &1, Core.Error, Core.Dynamic & Core.Dynamic>) ->
1348  Tape & Result<&2, &2, Error, Nat>:
1349  match result:
1350    case Fail{error}:
1351      (Tape{entries}, Fail{TensorError{error}})
1352    case Done{pair}:
1353      (input, output) = pair
1354      push(Sum{a, shape}, output, Unsaved{}, entries)