def ce_put

bend-ml-tensor-fork@0.1.6.0 · main.bend:345

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

345def ce_put(+i: U32, lab: Array<U32>, t: Array<F32>, q: TA.L2) -> B3<F32, U32, F32>:
346  match q:
347    case TA.L2{+x, z}:
348      B3{z, lab, Array.set(F32, t, i, x)}