jax.numpy.tanh

jax.numpy.tanh(x, /)

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

numpy.tanh 的 JAX 实现。

双曲正切定义为

\[tanh(x) = \frac{sinh(x)}{cosh(x)} = \frac{e^x - e^{-x}}{e^x + e^{-x}}\]

参数:

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

返回:

一个数组，包含 x 的每个元素的双曲正切，并提升为非精确的 dtype。

返回类型:

Array

注意

jnp.tanh 等价于计算 -1j * jnp.tan(1j * x)。

另请参阅

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

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

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

示例

>>> x = jnp.array([[-1, 0, 1],
...                [3, -2, 5]])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.tanh(x)
Array([[-0.762,  0.   ,  0.762],
       [ 0.995, -0.964,  1.   ]], dtype=float32)
>>> with jnp.printoptions(precision=3, suppress=True):
...   -1j * jnp.tan(1j * x)
Array([[-0.762+0.j,  0.   -0.j,  0.762-0.j],
       [ 0.995-0.j, -0.964+0.j,  1.   -0.j]],      dtype=complex64, weak_type=True)

对于复数值输入

>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.tanh(2-5j)
Array(1.031+0.021j, dtype=complex64, weak_type=True)
>>> with jnp.printoptions(precision=3, suppress=True):
...   -1j * jnp.tan(1j * (2-5j))
Array(1.031+0.021j, dtype=complex64, weak_type=True)