law sgd_params_in_order

bendtorch-core@0.1.0.0 · src/autograd/sgd_LAWS.bend:116

import Base
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as Backward
import bendtorch-core@0.1.0.0/src/autograd/sgd.bend as Sgd

116law sgd_params_in_order:
117  for +lr: F32
118  for +count: Nat
119  for +id: Nat
120  for rest: List<&2, Nat>
121  for entries: List<&1, Backward.Entry>
122  {Sgd.descend_all(lr, count, id <> rest, entries) ==
123    Sgd.continued(after => Sgd.descend_all(lr, count, rest, after), Sgd.descend_at(lr, entries, count, id, Backward.at_head(count, id))) :
124    Sgd.Walked()}