def weighted

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

import bend-ml-llama@0.2.1.0/main.bend as BendMlLlama

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

763def weighted(+hd: Nat, +t: Nat, v: Array<F32>, +w: List<&2, F32>) -> WR:
764  wr_read(hd, TA.gemm(1n, hd, 1n+t, U32.from_nat(1n+t), 1, U32.from_nat(hd), 1, TA.fill_list(w, 0, Array.new(F32, TA.cap_depth(1n+t), 0.0)), v, Array.new(F32, TA.cap_depth(hd), 0.0)))

Uses (11)

Used by (1)

bend-ml-llama@0.2.1.0/main:head_sr