mlx.core.arctan2

Contents

mlx.core.arctan2#

arctan2(a: scalar | array, b: scalar | array, /, *, stream: StreamOrDevice = None) → array#

Element-wise inverse tangent of the ratio of two arrays.

Parameters:
  • a (array or scalar) – Input array or scalar.

  • b (array or scalar) – Input array or scalar.

Returns:

The inverse tangent of the ratio of a and b.

Return type:

array