def acc_add

bend-ml-autograd@0.1.1.0 · main.bend:161

import Base

161# adds v to the gradient of variable i
162def acc_add(acc: List<&2, F32>, +i: Nat, +v: F32) -> List<&2, F32>:
163  match acc i:
164    case Nil{} _:
165      Nil{}
166    case Con{+h, t} 0n:
167      (h + v : F32) <> t
168    case Con{h, t} 1n+p:
169      h <> acc_add(t, p, v)