small modif to tf.gradients in tensorflow backend

j-zarp · Nov 23, 2016 · 38f5750 · 38f5750
1 parent d174ddf
commit 38f5750
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras/backend/tensorflow_backend.py b/keras/backend/tensorflow_backend.py
@@ -989,11 +989,11 @@ def function(inputs, outputs, updates=[], **kwargs):
     return Function(inputs, outputs, updates=updates)
 
 
-def gradients(loss, variables):
+def gradients(loss, variables, initial_vals=None):
     '''Returns the gradients of `variables` (list of tensor variables)
     with regard to `loss`.
     '''
-    return tf.gradients(loss, variables)
+    return tf.gradients(loss, variables, initial_vals)
 
 
 def stop_gradient(variables):