def Opt.mom_fin
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA
424def Opt.mom_fin(-r: Nat, -c: Nat, q: Q3) -> MatStep<r, c>:
425 match q:
426 case Q3{w, g, m}: MatStep{TA.Mat{w}, TA.Mat{g}, OneMom{TA.Mat{m}}}