def pack_result

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

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

166def pack_result(+shape: Shape.Shape,
167  result: Result<&2, &1, Core.Error, Core.Tensor<shape> & Core.Tensor<scalar_shape()>>) ->
168  Result<&2, &1, Core.Error, Core.Dynamic & Core.Dynamic>:
169  match result:
170    case Fail{error}:
171      Fail{error}
172    case Done{pair}:
173      (tensor, scalar) = pair
174      Done{(Core.pack(shape, tensor), Core.pack(scalar_shape(), scalar))}