def fork

bendtorch-core@0.1.0.0 · src/tensor/parallel.bend:65

import Base

65def fork(~A: Type, ~B: Type, pair: (A & B) & (A & B), low: A -> B -> Array<F32>, high: A -> B -> Array<F32>) -> Array<F32>:
66  (first, second) = pair
67  (l1, r1) = first
68  (l2, r2) = second
69  a b = low(l1, r1) high(l2, r2)
70  ANode{a, b}