jax.lax.atan2

jax.lax.atan2(x, y)

Elementwise 双项反正切：\(\mathrm{atan}({x \over y})\)。

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

参数:

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

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

返回:

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

返回类型:

Array

另请参阅

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

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