mlx.core.isfinite

目录

mlx.core.isfinite#

isfinite(a: array, stream: None | Stream | Device = None) array#

返回一个布尔数组,指示哪些元素是有限的。

如果元素既不是无穷大也不是 NaN,则它是有限的。

参数:

a (array) – 输入数组。

返回值:

指示哪些元素是有限的布尔数组。

返回类型:

array