python - 在更新 dict 值时累积变量？（尝试实现SGD+动量）

Question

所以我为香草 SGD 实现了以下字典：

        update_weights = dict(zip(weight_keys,
                                [grad_weight[key] -
                                 lr * convert_to_tensor(dx[key])]) for key in weight_keys)

我正在尝试用动量实现类似的东西，但是，我不确定如何能够累积速度项，以便我可以在列表理解术语中一次更新它：

v_(i) = mu * v_(i-1) - lr * convert_to_tensor(dx[key])
grad_weight[key] += v_i

有谁知道我如何使用列表理解（最好使用 Tensorflow）来做到这一点？

for key in weight_keys:
     v = mu * v - lr * convert_to_tensor(dx[key])
     update_weights[key] = grad_weight[key] + v