mlx.core.isinf

mlx.core.isinf(a: array, stream: Union[None, Stream, Device] = None) → array

Return a boolean array indicating which elements are +/- inifnity.

Parameters:

a (array) – Input array.

Returns:

The boolean array indicating which elements are +/- infinity.

Return type:

array