def as_sum

bendtorch-core@0.1.0.0 · src/autograd/sum_dims_LAWS.bend:16

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

16# reduce_to's result as a sum's: its error as a ReduceError.
17def as_sum(pair: Core.Dynamic & Result<&2, &1, Reduce.Error, Core.Dynamic>) -> Core.Dynamic & Result<&2, &1, SumDims.Error, Core.Dynamic>:
18  (input, result) = pair
19  match result:
20    case Fail{error}:
21      (input, Fail{SumDims.ReduceError{error}})
22    case Done{output}:
23      (input, Done{output})