Function gradcheck

extra/gradcheck.py:48–51

(func, input, eps = 1e-3, atol = 1e-3, rtol = 1e-3)

  return NJ

def gradcheck(func, input, eps = 1e-3, atol = 1e-3, rtol = 1e-3):
  NJ = numerical_jacobian(func, input, eps)
  J = jacobian(func, input)
  return np.allclose(J, NJ, atol = atol, rtol = rtol)