реализация resource_apply_keras_momentum с помощью Python?

Question

Ищу реализацию tensorflow.python.training_ops.resource_apply_keras_momentum. Обычно это используется в импульсе NAg:

1. v_t+1 = momentum * v_t - lr*grad(theta_t + momentum *v_t)

2. theta_t+1 = theta_t + v_t+1

На самом деле, я хотел бы знать, как оценить градиент на v_t.

Кто-нибудь может мне помочь?