def brow_leaf2

bend-ml-tensor-array@0.1.19.0 · main.bend:1725

import Base

1725# n numbers starting at flat index i of one band (a band may have no rows)
1726def brow_leaf2(-r: Nat, -c: Nat, +n: Nat, +i: U32, d: Array<F32>) -> BRow<r, c>:
1727  match n:
1728    case 0n:
1729      BRow{BLeaf{Mat{d}}, Nil{}}
1730    case 1n+p:
1731      brow_fin(r, c, read_l(p, ((i + U32.from_nat(n) : U32) - 1 : U32), Nil{}, Array.get(F32, d, ((i + U32.from_nat(n) : U32) - 1 : U32))))