def get_bands

bend-ml-llama@0.2.1.0 · main.bend:1651

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

1651# tensor t (w/st.txt's order) as bands of depth d
1652def get_bands(+dir: String, +st: List<&2, U32>, +t: Nat, +d: Nat, +r: Nat, +c: Nat) -> IO(TA.Bands<r, c>):
1653  do IO<TA.Bands<r, c>>:
1654    check_span(st_path(dir), st_hi(st, t), st_lo(st, t), Nat.mul(r, c), t)
1655    load_tree(c, r, d, r, {==}, st_path(dir), st_lo(st, t), 0n)