Parallel.fork_balanced: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel
Lines 157-164 of bendtorch-core@0.1.0.0/src/tensor/parallel.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&start_line=157&end_line=164

import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model

def fork_balanced(~A: Type, ~B: Type, +inner: Nat, pair: (A & B) & (A & B), -low: A -> B -> Array<F32>, -high: A -> B -> Array<F32>,
  low_valid: @l: A -> @r: B -> Model.Balanced(F32, inner, low(l, r)),
  high_valid: @l: A -> @r: B -> Model.Balanced(F32, inner, high(l, r))) ->
  Model.Balanced(F32, 1n+inner, fork(~A, ~B, pair, low, high)):
  (first, second) = pair
  (l1, r1) = first
  (l2, r2) = second
  (low_valid(l1, r1), high_valid(l2, r2))

Uses (7):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Array
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Sigma
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/storage/model.bend&name=Balanced
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=fork