def mom_fin

bendtorch-gpu@1.3.0.0 · main.bend:254

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA
import bend-ml-tensor-gpu@0.1.5.0/main.bend as TG
import bendtorch@1.3.0.0/main.bend as T

254def mom_fin(-r: Nat, -c: Nat, q: TG.R3) -> T.MatStep<r, c>:
255  match q:
256    case TG.R3{x1, x2, x3, x4, TG.O3{w, g, m}}: T.MatStep{TA.Mat{w}, TA.Mat{g}, T.OneMom{TA.Mat{m}}}

Uses (6)

Used by (1)

bendtorch-gpu@1.3.0.0/main:opt_mat