def MappingWith

bendtorch-core@0.1.0.0 · src/autograd/reduce_LAWS.bend:73

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
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality
import bendtorch-core@0.1.0.0/src/autograd/reduce.bend as Reduce

73def MappingWith(-input: Core.Dynamic, +target: Shape.Shape, plan: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>,
74  pair: Core.Dynamic & Result<&2, &1, Reduce.Error, Core.Dynamic>) -> Type:
75  (retained, result) = pair
76  match result:
77    case Fail{error}:
78      Unit
79    case Done{output}:
80      {Totality.dimensions(output) == Shape.dimensions(target) : List<&2, U32>} &
81        (@+values: List<&2, F32> -> @-listed: Listed(input, values) ->
82          @+index: U32 -> @inside: {Nat.is_lt(U32.to_nat(index), Shape.logical_numel(Shape.dimensions(target))) == True{} : Bool} ->
83            {Core.checked_observe_dynamic(output, index) ==
84              planned_fiber(values, index, Shape.logical_numel(Totality.dimensions(input)), plan) : Maybe<&2, F32>})