diff --git a/jax/_src/numpy/ufuncs.py b/jax/_src/numpy/ufuncs.py index f20bab9b1b89..b2b7f5e710c3 100644 --- a/jax/_src/numpy/ufuncs.py +++ b/jax/_src/numpy/ufuncs.py @@ -1015,6 +1015,7 @@ def arctanh(x: ArrayLike, /) -> Array: For complex-valued input: + >>> x1 = jnp.array([-2+0j, 3+0j, 4-1j]) >>> with jnp.printoptions(precision=3, suppress=True): ... jnp.arctanh(x1) Array([-0.549+1.571j, 0.347+1.571j, 0.239-1.509j], dtype=complex64)