jax.lax.atan2

jax.lax.atan2(x, y)

Elementwise two-term arc tangent: \(\mathrm{atan}({x \over y})\).

This function lowers directly to the stablehlo.atan2 operation.

Parameters:

• x (ArrayLike) – input arrays. Must have a matching floating-point or complex dtypes. If neither is a scalar, the two arrays must have the same number of dimensions and be broadcast-compatible.

• y (ArrayLike) – input arrays. Must have a matching floating-point or complex dtypes. If neither is a scalar, the two arrays must have the same number of dimensions and be broadcast-compatible.

Returns:

Array of the same shape and dtype as x and y containing the element-wise arc tangent of \(x \over y\), respecting the quadrant indicated by the sign of each input.

Return type:

Array

See also

• jax.lax.tan(): elementwise tangent.

• jax.lax.atan(): elementwise one-term arc tangent.