def pq_b
import bendtorch-fast@1.3.0.0/main.bend as BendtorchFastimport Base
import bend-ml-tensor-fork@0.1.5.0/main.bend as TF
import bendtorch@1.3.0.0/main.bend as T
100def pq_b(q: T.PQ) -> TF.B4<F32, F32, U32, U32>:
101 match q:
102 case T.PQ{w, g, m, v}: TF.B4{w, g, m, v}