def test_reverse(): def fun(x): return x[0] + x[1]**2 + x[2]**3 htrue = np.array([0., 2., 18.]) h_fun = nd.Hessdiag(fun, method='reverse') hd = h_fun([1, 2, 3]) assert_allclose(hd, htrue)
def test_reverse(): def fun(x): return x[0] + x[1] ** 2 + x[2] ** 3 htrue = np.array([0., 2., 18.]) Hfun = nd.Hessdiag(fun, method='reverse') hd = Hfun([1, 2, 3]) _error = hd - htrue assert_array_almost_equal(hd, htrue)
def test_forward(): def fun(x): return x[0] + x[1]**2 + x[2]**3 htrue = np.array([0., 2., 18.]) h_fun = nd.Hessdiag(fun) hd = h_fun([1, 2, 3]) assert_allclose(hd, htrue)