def test_logsumexp6():
x = npr.randn(1, 5)
def f(a):
return autograd.scipy.misc.logsumexp(a, axis=1, keepdims=True)
check_grads(f, x)
check_grads(lambda a: to_scalar(grad(f)(a)), x)
def test_logsumexp6():
x = npr.randn(1, 5)
def f(a):
return autograd.scipy.misc.logsumexp(a, axis=1, keepdims=True)
check_grads(f, x)
check_grads(lambda a: to_scalar(grad(f)(a)), x)
def test_yn():
x = npr.randn()**2 + 0.3
fun = lambda x: to_scalar(autograd.scipy.special.yn(2, x))
d_fun = grad(fun)
check_grads(fun, x)
check_grads(d_fun, x)
def test_polygamma():
x = npr.randn()
fun = lambda x: to_scalar(autograd.scipy.special.polygamma(0, x))
d_fun = grad(fun)
check_grads(fun, x)
check_grads(d_fun, x)
def fun(B):
return to_scalar(
spla.solve_triangular(A, B, trans=trans, lower=lower))
def fun(A):
return to_scalar(spla.sqrtm(A))
def test_yn():
x = npr.randn()**2 + 0.3
fun = lambda x: to_scalar(autograd.scipy.special.yn(2, x))
d_fun = grad(fun)
check_grads(fun, x)
check_grads(d_fun, x)
def test_polygamma():
x = npr.randn()
fun = lambda x: to_scalar(autograd.scipy.special.polygamma(0, x))
d_fun = grad(fun)
check_grads(fun, x)
check_grads(d_fun, x)
