mlx.core.isinf

isinf(a: array, stream: None | Stream | Device = None) → array

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

參數:

a (array) -- 輸入陣列。

回傳值:

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

回傳值型別:

array