def merge

bendtorch-core@0.1.0.0 · src/autograd/backward.bend:225

import Base

225def merge(entries: List<&2, Entry>, gradients: List<&2, Maybe<&2, F32>>) -> List<&2, Entry>:
226  match entries gradients:
227    case Nil{} gradients:
228      Nil{}
229    case Con{entry, rest} Nil{}:
230      entry <> rest
231    case Con{Entry{node, value, slot}, rest} Con{gradient, more}:
232      Entry{node, value, accumulate(slot, gradient)} <> merge(rest, more)