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)