Module ImportFrom alias Import alias identifier Import alias identifier ImportFrom alias ImportFrom alias alias alias alias alias alias alias alias alias alias ImportFrom alias Expr Call AttributeLoad NameLoad attr Num FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr BinOpPow Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call NameLoad NameLoad Assign NameStore Call NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr Num Expr Call NameLoad Call NameLoad NameLoad SubscriptLoad Call NameLoad NameLoad Index Num Expr Call NameLoad Call NameLoad NameLoad SubscriptLoad Call NameLoad NameLoad Index Num decorator_list FunctionDef arguments args defaults body Assign NameStore Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad FunctionDef arguments args NameParam defaults body Return Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad decorator_list Assign NameStore Call NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Expr Call NameLoad Call NameLoad NameLoad BinOpAdd NameLoad AttributeLoad NameLoad attr decorator_list FunctionDef arguments args defaults body FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam NameParam NameParam defaults Num Num body Return BinOpAdd BinOpAdd BinOpAdd BinOpAdd BinOpAdd BinOpAdd NameLoad Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Return Call NameLoad NameLoad NameLoad NameLoad NameLoad NameLoad keyword NameLoad keyword NameLoad decorator_list Assign NameStore Num Assign NameStore Num Assign NameStore Num Assign NameStore Num Assign NameStore Num Assign NameStore Num Assign NameStore Num Assign NameStore Call Call NameLoad NameLoad keyword ListLoad Num Num NameLoad NameLoad NameLoad NameLoad NameLoad keyword NameLoad keyword NameLoad Assign NameStore Call NameLoad NameLoad NameLoad NameLoad Expr Call NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam NameParam defaults Call AttributeLoad NameLoad attr BinOpAdd NameLoad Call AttributeLoad NameLoad attr NameLoad Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr BinOpAdd SubscriptLoad NameLoad Index Num Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index Num Assign TupleStore NameStore NameStore TupleLoad Call AttributeLoad NameLoad attr Num Call AttributeLoad NameLoad attr Num Expr Call NameLoad Call Call NameLoad NameLoad NameLoad NameLoad Call Call NameLoad NameLoad TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body FunctionDef arguments args NameParam defaults body Return BinOpAdd BinOpAdd NameLoad Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr NameLoad decorator_list Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Call Call NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr ListComp Call NameLoad NameLoad SubscriptLoad NameLoad Index NameLoad comprehension NameStore Call NameLoad Call NameLoad NameLoad Expr Call NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body FunctionDef arguments args NameParam NameParam defaults body Return BinOpAdd BinOpAdd NameLoad Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr NameLoad decorator_list Assign NameStore Num Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Num Assign NameStore Call Call NameLoad NameLoad keyword NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr ListComp SubscriptLoad Call NameLoad NameLoad NameLoad SubscriptLoad NameLoad Index NameLoad Index NameLoad comprehension NameStore Call NameLoad Call NameLoad NameLoad Expr Call NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr Num Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr Num Num Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad keyword Call AttributeLoad NameLoad attr NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Call AttributeLoad NameLoad attr Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Lambda arguments args NameParam defaults Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Num Assign NameStore Call AttributeLoad NameLoad attr Num Num Assign NameStore Call Call NameLoad NameLoad NameLoad Expr Call NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad keyword Call AttributeLoad NameLoad attr NameLoad Call Call NameLoad NameLoad NameLoad NameLoad decorator_list
N/A __future__ absolute_import N/A autograd.numpy np N/A autograd.numpy.random npr autograd.util * autograd grad elementwise_grad jacobian value_and_grad grad_and_aux hessian_vector_product hessian multigrad jacobian vector_jacobian_product builtins range N/A N/A N/A npr seed 1 test_value_and_grad N/A N/A N/A N/A N/A fun N/A N/A N/A x N/A N/A N/A np sum N/A N/A N/A np sin x 2 N/A dfun N/A grad fun N/A dfun_both N/A value_and_grad fun N/A x N/A N/A npr randn 5 N/A N/A check_equivalent N/A fun x N/A N/A dfun_both x N/A 0 N/A N/A check_equivalent N/A dfun x N/A N/A dfun_both x N/A 1 N/A test_hessian N/A N/A N/A N/A N/A D 5 N/A H N/A N/A npr randn D D fun N/A N/A x N/A N/A N/A N/A N/A np dot N/A N/A np dot x H x N/A N/A hess N/A hessian fun N/A x N/A N/A npr randn D N/A N/A check_equivalent N/A hess x N/A H N/A H T N/A test_multigrad N/A N/A N/A N/A complicated_fun N/A N/A a b c d e f g N/A 1.1 9.0 N/A N/A N/A N/A N/A N/A N/A N/A a N/A N/A np sin b N/A N/A np cosh c N/A N/A np cos d N/A N/A np tan e f g N/A complicated_fun_3_1 N/A N/A d b N/A N/A N/A N/A complicated_fun A b C d E f F g G N/A N/A A 0.5 N/A B -0.3 N/A C 0.2 N/A D -1.1 N/A E 0.7 N/A F 0.6 N/A G -0.1 N/A exact N/A N/A multigrad complicated_fun argnums N/A 3 1 A B C D E f F g G N/A numeric N/A nd complicated_fun_3_1 D B N/A N/A check_equivalent exact numeric N/A test_multigrad_onearg N/A N/A N/A N/A N/A fun N/A N/A N/A x y N/A N/A N/A np sum N/A x N/A N/A np sin y N/A packed_fun N/A N/A N/A xy N/A N/A N/A np sum N/A N/A xy N/A 0 N/A N/A np sin N/A xy N/A 1 N/A N/A A B N/A N/A N/A npr randn 3 N/A N/A npr randn 3 N/A N/A check_equivalent N/A N/A multigrad fun A B N/A N/A grad packed_fun N/A A B N/A test_elementwise_grad N/A N/A N/A N/A simple_fun N/A N/A a N/A N/A N/A N/A N/A a N/A N/A np sin a N/A N/A np cosh a N/A N/A A N/A N/A npr randn 10 N/A exact N/A N/A elementwise_grad simple_fun A N/A numeric N/A N/A np squeeze N/A N/A np array N/A N/A nd simple_fun N/A A N/A i N/A i N/A range N/A len A N/A N/A check_equivalent exact numeric N/A test_elementwise_grad_multiple_args N/A N/A N/A N/A simple_fun N/A N/A a b N/A N/A N/A N/A N/A a N/A N/A np sin a N/A N/A np cosh b N/A N/A A 0.9 N/A B N/A N/A npr randn 10 N/A argnum 1 N/A exact N/A N/A elementwise_grad simple_fun argnum argnum A B N/A numeric N/A N/A np squeeze N/A N/A np array N/A N/A N/A nd simple_fun A N/A B N/A i N/A argnum N/A i N/A range N/A len B N/A N/A check_equivalent exact numeric N/A test_hessian_vector_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np sum N/A N/A np sin a N/A a N/A N/A npr randn 5 N/A v N/A N/A npr randn 5 N/A H N/A N/A hessian fun a N/A N/A check_equivalent N/A N/A np dot H v N/A N/A hessian_vector_product fun a v N/A test_hessian_matrix_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np sum N/A N/A np sin a N/A a N/A N/A npr randn 5 4 N/A V N/A N/A npr randn 5 4 N/A H N/A N/A hessian fun a N/A N/A check_equivalent N/A N/A np tensordot H V N/A N/A hessian_vector_product fun a V N/A test_hessian_tensor_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np sum N/A N/A np sin a N/A a N/A N/A npr randn 5 4 3 N/A V N/A N/A npr randn 5 4 3 N/A H N/A N/A hessian fun a N/A N/A check_equivalent N/A N/A np tensordot H V axes N/A N/A np ndim V N/A N/A hessian_vector_product fun a V N/A test_vector_jacobian_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np roll N/A N/A np sin a 1 N/A a N/A N/A npr randn 5 N/A V N/A N/A npr randn 5 N/A J N/A N/A jacobian fun a N/A N/A check_equivalent N/A N/A np dot N/A V T J N/A N/A vector_jacobian_product fun a V N/A test_matrix_jacobian_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np roll N/A N/A np sin a 1 N/A a N/A N/A npr randn 5 4 N/A V N/A N/A npr randn 5 4 N/A J N/A N/A jacobian fun a N/A N/A check_equivalent N/A N/A np tensordot V J N/A N/A vector_jacobian_product fun a V N/A test_tensor_jacobian_product N/A N/A N/A N/A N/A fun N/A N/A N/A a N/A N/A N/A np roll N/A N/A np sin a 1 N/A a N/A N/A npr randn 5 4 3 N/A V N/A N/A npr randn 5 4 N/A J N/A N/A jacobian fun a N/A N/A check_equivalent N/A N/A np tensordot V J axes N/A N/A np ndim V N/A N/A vector_jacobian_product fun a V N/A