mindspore.numpy.array_equal

查看源文件
mindspore.numpy.array_equal(a1, a2, equal_nan=False)[源代码]

当输入数组shape相同且所有元素相等时,返回 True

说明

在MindSpore中,会返回一个bool tensor,因为在图模式下,该值无法在编译时追踪和计算。由于Ascend平台对 nan 的处理不同,目前在Ascend上不支持 equal_nan 参数。

参数:
  • a1/a2 (Union[int, float, bool, list, tuple, Tensor]) - 输入数组。

  • equal_nan (bool,可选) - 是否将 NaN 视为相等。默认值: False

返回:

标量bool tensor,如果输入相等,值为 True ,否则为 False

异常:
  • TypeError - 如果输入参数非上述给定的类型。

支持平台:

GPU CPU Ascend

样例:

>>> import mindspore.numpy as np
>>> a = [0,1,2]
>>> b = [[0,1,2], [0,1,2]]
>>> print(np.array_equal(a,b))
False