def read_f32_at

bendtorch-load@1.3.0.0 · main.bend:141

import Base
import bend-ml-binary-read@0.2.0.0/main.bend as R
import bendtorch@1.3.0.0/main.bend as T

141# n little-endian F32 numbers at byte `off` of a file of any size (a tensor inside a larger file)
142def read_f32_at(+path: String, off: W64, +n: U32) -> IO(Array<F32>):
143  match off:
144    case W64{hi, lo}:
145      do IO<Array<F32>>:
146        f : File <- IO.try(File, File.open(path, "r"))
147        rr : File & (Array<F32> & Result<&1, &1, U32 & String, U32>) <- R.read_f32_at(f, hi, lo, n, Array.new(F32, T.depth(U32.to_nat(n)), 0.0))
148        read_done(path, n, rr)