jax.numpy.arctan

jax.numpy.arctan(x, /)

计算输入的反正切值（元素级）。

JAX 实现的 numpy.arctan。

参数:

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

返回:

一个数组，包含每个元素 x 的反正切值（弧度），范围在 [-pi/2, pi/2]，提升为非精确数据类型。

返回类型:

Array

注意

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

另请参阅

• jax.numpy.tan()：计算输入中每个元素的三角正切值。

• jax.numpy.arcsin() 和 jax.numpy.asin()：计算输入中每个元素的三角正弦的逆值。

• jax.numpy.arccos() 和 jax.numpy.atan()：计算输入中每个元素的三角余弦的逆值。

示例

>>> x = jnp.array([-jnp.inf, -20, -1, 0, 1, 20, jnp.inf])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.arctan(x)
Array([-1.571, -1.521, -0.785,  0.   ,  0.785,  1.521,  1.571], dtype=float32)

对于复数值输入

>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.arctan(2+7j)
Array(1.532+0.133j, dtype=complex64, weak_type=True)