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)