mlx.core.isneginf

Contents

mlx.core.isneginf#

isneginf(a: array, stream: None | Stream | Device = None) array#

Return a boolean array indicating which elements are negative infinity.

Parameters:
  • a (array) – Input array.

  • stream (Union[None, Stream, Device]) – Optional stream or device.

Returns:

The boolean array indicating which elements are negative infinity.

Return type:

array