mlx.core.isnan

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

Return a boolean array indicating which elements are NaN.

参数:

a (array) -- Input array.

返回:

The boolean array indicating which elements are NaN.

返回类型:

array