jax.numpy.arccosh

jax.numpy.arccosh(x, /)

计算输入的双曲余弦的逐元素反函数。

JAX 对 numpy.arccosh 的实现。

双曲余弦的反函数定义为：

\[arccosh(x) = \ln(x + \sqrt{x^2 - 1})\]

参数:

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

返回值:

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

返回类型:

数组

注意

• 对于 [-inf, 1) 范围内的实数值，jnp.arccosh 返回 nan。

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

另请参阅

• jax.numpy.cosh(): 计算输入的逐元素双曲余弦。

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

• jax.numpy.arctanh(): 计算输入的逐元素双曲正切的反函数。

示例

>>> x = jnp.array([[1, 3, -4],
...                [-5, 2, 7]])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.arccosh(x)
Array([[0.   , 1.763,   nan],
       [  nan, 1.317, 2.634]], dtype=float32)

对于复数值输入

>>> x1 = jnp.array([-jnp.inf+0j, 1+2j, -5+0j])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.arccosh(x1)
Array([  inf+3.142j, 1.529+1.144j, 2.292+3.142j], dtype=complex64)