def mat_step_fin

bendtorch@1.3.0.0 · main.bend:1533

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

1533# one optimizer step on a layer whose parameter is a single matrix
1534def mat_step_fin(-r: Nat, -c: Nat, q: MatStep<r, c>) -> Step<TA.Mat<r, c>>:
1535  match q:
1536    case MatStep{w, g, st}: Step{w, g, st}