jax.numpy.arctanh

jax.numpy.arctanh(x, /)

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

JAX 对 numpy.arctanh 的实现。

双曲正切的反函数定义为：

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

参数:

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

返回:

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

返回类型:

Array

注意

• jnp.arctanh 对于在 [-1, 1] 范围外的实数值返回 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)