def step

bendtorch-core@0.1.0.0 · src/tensor/contracts/fold.bend:20

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

20def step(~O: Data, ~g: O -> F32 -> F32 -> F32, +operation: O, +kept: Bool,
21  rec: Core.Dynamic -> F32 -> Core.Dynamic & Result<&2, &1, Core.Error, F32>,
22  accumulator: F32, pair: Core.Dynamic & Result<&2, &2, Core.Error, F32>) -> Core.Dynamic & Result<&2, &1, Core.Error, F32>:
23  (input, result) = pair
24  match result:
25    case Fail{error}:
26      (input, Fail{error})
27    case Done{x}:
28      rec(input, chosen(~O, ~g, operation, kept, accumulator, x))