~/jendhub

def copy_leaf

bendtorch-fast@1.3.0.0 · main.bend:215

import bendtorch-fast@1.3.0.0/main.bend as BendtorchFast
import Base
import bend-ml-tensor-fork@0.1.5.0/main.bend as TF
import bendtorch@1.3.0.0/main.bend as T

215# numbers i0 .. i0+cnt-1 of a into b
216def copy_leaf(+e: Unit, +i0: U32, +cnt: Nat, r: TF.B2<F32, F32>) -> TF.B2<F32, F32>:
217  match cnt r:
218    case 0n r:
219      r
220    case 1n+k TF.B2{a, b}:
221      a2_b2(T.copy_n(k, i0, i0, b, Array.get(F32, a, i0)))

Uses (8)

Used by (1)