mlx.core.isinf

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

返回一个布尔数组，指示哪些元素是 +/- 无穷大。

参数:

a (array) – 输入数组。

返回值:

指示哪些元素是 +/- 无穷大的布尔数组。

返回类型:

数组