def mat_step

bendtorch-gpu@1.3.0.0 · main.bend:275

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

275def mat_step(+r: Nat, +c: Nat, +op: T.Opt, p: TA.Mat<r, c>, g: TA.Mat<r, c>, st: T.OptState<TA.Mat<r, c>>) -> T.Step<TA.Mat<r, c>>:
276  T.mat_step_fin(r, c, opt_mat(r, c, op, p, g, st))