def descend_all

bendtorch-core@0.1.0.0 · src/autograd/sgd.bend:148

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

148# Each parameter in list order, each on the entries the one before it left.
149def descend_all(+lr: F32, +count: Nat, params: List<&2, Nat>, entries: List<&1, Backward.Entry>) -> Walked():
150  match params:
151    case Nil{}:
152      (entries, Done{Unit{}})
153    case Con{+id, rest}:
154      continued(entries => descend_all(lr, count, rest, entries), descend_at(lr, entries, count, id, Backward.at_head(count, id)))