def Bands.fill_at

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

import Base

1640# writes the n numbers of xs into b starting at flat index i (row-major, as Mat.fill_at);
1641# numbers past r*c are dropped. For loading a large matrix block by block.
1642def Bands.fill_at(+r: Nat, +c: Nat, +i: Nat, +n: Nat, +xs: List<&2, F32>, b: Bands<r, c>) -> Bands<r, c>:
1643  bfill_top(Nat.is_le(Nat.add(i, n), Nat.mul(r, c)), r, c, b, i, n, xs)