def Obs

bendtorch-core@0.1.0.0 · src/autograd/reduce_parallel_proof.bend:244

import Base

244# Each word is what A holds at its position.
245def Obs(+d: Nat, -A: Array<F32>, positions: List<&2, Nat>, words: List<&2, F32>) -> Data:
246  match positions words:
247    case Nil{} Nil{}:
248      Unit
249    case p <> ps w <> ws:
250      ObsStep<d, A, p, w, ps, ws>
251    case Nil{} w <> ws:
252      Empty
253    case p <> ps Nil{}:
254      Empty