Exemplo n.º 1
0
 def test_jvp_linearized(self, f, args):
   jtu.check_jvp(f, partial(jvp_unlinearized, f), args,
                 rtol={np.float32: 3e-2})
Exemplo n.º 2
0
 def test_jvp(self, f, args):
   jtu.check_jvp(f, partial(jvp, f), args, rtol={np.float32: 3e-2})