def get_bands

bend-ml-llama@0.2.0.0 · main.bend:1255

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

1255def get_bands(+dir: String, +name: String, +d: Nat, +r: Nat, +c: Nat) -> IO(TA.Bands<r, c>):
1256  do IO<TA.Bands<r, c>>:
1257    check_size(wpath(dir, name), Nat.mul(r, c))
1258    load_tree(c, r, TA.Bands.zeros(d, r, c), r, {==}, wpath(dir, name), 0n)