def Mat.store

bendtorch@1.3.0.0 · main.bend:579

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

579def Mat.store(+r: Nat, +c: Nat, m: TA.Mat<r, c>, xs: Array<F32>, +off: U32) -> PA<TA.Mat<r, c>>:
580  match m:
581    case TA.Mat{ma}:
582      mat_store_fin(r, c, (off + U32.from_nat(Nat.mul(r, c)) : U32), store_cells(Nat.mul(r, c), 0, off, Array.get(F32, ma, 0), xs))