mlx.core.isclose

目录

mlx.core.isclose#

isclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#

返回一个布尔数组，表示两个数组在容差范围内元素是否相等。

无穷大值如果符号相同则视为相等，NaN 值则视为不相等，除非 equal_nan 为 True。

如果满足以下条件，则认为两个值相等：

abs(a - b) <= (atol + rtol * abs(b))

注意，与 array_equal() 不同，此函数支持 NumPy 风格的广播。

参数：

  • a (array) – 输入数组。

  • b (array) – 输入数组。

  • rtol (float) – 相对容差。

  • atol (float) – 绝对容差。

  • equal_nan (bool) – 如果为 True，则将 NaNs 视为相等。默认为 False。

返回值：

布尔类型的输出标量，表示数组是否接近。

返回类型：

array

目录