def reshape_as_sum

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

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

25def reshape_as_sum(result: Result<&2, &1, Transforms.Error, Core.Dynamic>) -> Result<&2, &1, SumDims.Error, Core.Dynamic>:
26  match result:
27    case Fail{error}:
28      Fail{SumDims.TransformError{error}}
29    case Done{output}:
30      Done{output}