jax.lax.atan2

jax.lax.atan2(x, y)

逐元素计算两项反正切：\(\mathrm{atan}({x \over y})\)。

此函数直接降到 stablehlo.atan2 操作。

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的浮点型或复数型 dtypes。如果两者都不是标量，则两个数组必须具有相同的维数并且是广播兼容的。

• y (ArrayLike) – 输入数组。必须具有匹配的浮点型或复数型 dtypes。如果两者都不是标量，则两个数组必须具有相同的维数并且是广播兼容的。

返回:

形状和 dtype 与 x 和 y 相同的数组，包含 \(x \over y\) 的逐元素反正切，尊重每个输入的符号指示的象限。

返回类型:

Array

另请参阅

• jax.lax.tan()：逐元素正切。

• jax.lax.atan()：逐元素计算单项反正切。