Method tanh

tests/test_jax.py:14–16

(self, x)

class TestJAX(unittest.TestCase):
    def tanh(self, x):
        y = np.exp(-2.0 * x)
        return (1.0 - y) / (1.0 + y)

    def test_grad(self):
        grad_tanh = grad(self.tanh)