def reshaped

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

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

32# A sum's result with its output reshaped to `dimensions`.
33def reshaped(+dimensions: List<&2, U32>, pair: Core.Dynamic & Result<&2, &1, SumDims.Error, Core.Dynamic>) ->
34  Core.Dynamic & Result<&2, &1, SumDims.Error, Core.Dynamic>:
35  (input, result) = pair
36  match result:
37    case Fail{error}:
38      (input, Fail{error})
39    case Done{output}:
40      (input, reshape_as_sum(Transforms.reshape_dynamic(output, dimensions)))