jax.numpy.arctanh

jax.numpy.arctanh(x, /)

计算输入的元素级双曲正切反函数。

numpy.arctanh 的 JAX 实现。

双曲正切的反函数定义为

\[arctanh(x) = \frac{1}{2} [\ln(1 + x) - \ln(1 - x)]\]

参数:

x (ArrayLike) – 输入数组或标量。

返回:

返回一个与 x 形状相同的数组，其中包含 x 中每个元素的双曲正切反函数值，并提升为非精确数据类型。

返回类型:

数组

注意

• 对于在 [-1, 1] 范围之外的实数值，jnp.arctanh 返回 nan。

• 对于复数输入，jnp.arctanh 遵循 numpy.arctanh 的分支切割约定。

另请参阅

• jax.numpy.tanh()：计算输入的逐元素双曲正切。

• jax.numpy.arcsinh()：计算输入的逐元素双曲正弦反函数。

• jax.numpy.arccosh()：计算输入的逐元素双曲余弦反函数。

示例

>>> x = jnp.array([-2, -1, -0.5, 0, 0.5, 1, 2])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.arctanh(x)
Array([   nan,   -inf, -0.549,  0.   ,  0.549,    inf,    nan], dtype=float32)

对于复数值输入

>>> 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)