Merge pull request #541 from titaneric/master

Update vjp and jvp of power
HIPS · Oct 6, 2019 · 12989c2 · 12989c2
2 parents 22d2c8c + 009cfeb
commit 12989c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/autograd/numpy/numpy_jvps.py b/autograd/numpy/numpy_jvps.py
@@ -49,7 +49,7 @@
 defjvp(anp.remainder,  lambda g, ans, x, y : broadcast(g, ans),
                        lambda g, ans, x, y : -g * anp.floor(x/y))
 defjvp(anp.power,      lambda g, ans, x, y : g * y * x ** anp.where(y, y - 1, 1.),
-                       lambda g, ans, x, y : g * anp.log(replace_zero(x, 1.)) * x ** y)
+                       lambda g, ans, x, y : g * anp.log(replace_zero(x, 1.)) * ans)
 defjvp(anp.arctan2,    lambda g, ans, x, y : g * y / (x**2 + y**2),
                        lambda g, ans, x, y : g * -x / (x**2 + y**2))
 

diff --git a/autograd/numpy/numpy_vjps.py b/autograd/numpy/numpy_vjps.py
@@ -57,7 +57,7 @@
                         lambda ans, x, y : unbroadcast_f(y, lambda g: -g * anp.floor(x/y)))
 defvjp(anp.power,
     lambda ans, x, y : unbroadcast_f(x, lambda g: g * y * x ** anp.where(y, y - 1, 1.)),
-    lambda ans, x, y : unbroadcast_f(y, lambda g: g * anp.log(replace_zero(x, 1.)) * x ** y))
+    lambda ans, x, y : unbroadcast_f(y, lambda g: g * anp.log(replace_zero(x, 1.)) * ans))
 defvjp(anp.arctan2,     lambda ans, x, y : unbroadcast_f(x, lambda g: g * y / (x**2 + y**2)),
                         lambda ans, x, y : unbroadcast_f(y, lambda g: g * -x / (x**2 + y**2)))
 defvjp(anp.hypot,