def sum_result

bendtorch-core@0.1.0.0 · src/tensor/reduction.bend:62

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

62def sum_result(-shape: Shape.Shape, result: Result<&2, &1, Core.Error, Core.Tensor<shape> & F32>) ->
63  Result<&2, &1, Core.Error, Core.Tensor<shape> & Core.Tensor<scalar_shape()>>:
64  match result:
65    case Fail{error}:
66      Fail{error}
67    case Done{pair}:
68      (tensor, value) = pair
69      scalar_result(shape, tensor, Core.new(scalar_shape(), value))