def mom_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
55def mom_b(q: T.Q3) -> TF.B3<F32, F32, F32>:
56 match q:
57 case T.Q3{w, g, m}: TF.B3{w, g, m}