def right_step

bendtorch-core@0.1.0.0 · src/tensor/contracts/pointwise.bend:25

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

25def right_step(~O: Data, ~f: O -> F32 -> F32 -> F32, +operation: O, -shape: Shape.Shape, +index: U32,
26  rec: State<shape> -> (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>,
27  left: Core.Dynamic, output: Core.Tensor<shape>, x: F32, pair: Core.Dynamic & Result<&2, &2, Core.Error, F32>) ->
28  (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>:
29  (right, result) = pair
30  match result:
31    case Fail{error}:
32      ((left, right), Fail{error})
33    case Done{y}:
34      write_step(shape, rec, left, right, Core.write_flat(shape, output, index, f(operation, x, y)))