jax.numpy.arctan

jax.numpy.arctan(x, /)

计算输入的反正切三角函数，按元素进行。

JAX 实现的 numpy.arctan。

参数:

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

返回:

一个数组，包含每个元素 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)