mlx.core.isposinf

Contents

mlx.core.isposinf#

isposinf(a: array, stream: StreamOrDevice = None) → array#

Return a boolean array indicating which elements are positive infinity.

Parameters:

  • a (array) – Input array.

  • stream (StreamOrDevice) – Optional stream or device.

Returns:

The boolean array indicating which elements are positive infinity.

Return type:

array

Contents