mlx.core.isnan

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

返回一个布尔数组，指示哪些元素是 NaN。

参数:

a (array) – 输入数组。

返回值:

布尔数组，指示哪些元素是 NaN。

返回类型:

array