def Bands.matvec

bend-ml-tensor-array@0.1.19.0 · main.bend:1703

import Base

1703# y = B · x for B (r x c, one row per output) and x (1 x c); returns B intact and y (1 x r).
1704# The same product as Mat.matmul_nt(1n, c, r, ..), and the same numbers, bit for bit.
1705def Bands.matvec(+r: Nat, +c: Nat, b: Bands<r, c>, x: Mat<1n, c>) -> BV<r, c>:
1706  match x:
1707    case Mat{xa}:
1708      bv_fin(r, c, bmv(c, r, b, r, {==}, Array.clone(F32, xa)))