mindspore.ops.EqualCount

查看源文件
class mindspore.ops.EqualCount[源代码]

计算两个Tensor的相同元素的数量。

两个输入Tensor必须具有相同的数据类型和shape。

输入:
  • x (Tensor) - 第一个输入Tensor。如果确定了 y 的数据类型和shape,则 x 必须与 y 相同,反之亦然。 \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。

  • y (Tensor) - 第二个输入Tensor。如果确定了 x 的数据类型和shape,则 y 必须与 x 相同,反之亦然。

输出:

Tensor,数据类型与输入Tensor相同,shape为 \((1,)\)

异常:
  • TypeError - 如果 xy 不是Tensor。

  • ValueError - 如果 xy 的shape不相等。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([1, 2, 3]), mindspore.int32)
>>> y = Tensor(np.array([1, 2, 4]), mindspore.int32)
>>> equal_count = ops.EqualCount()
>>> output = equal_count(x, y)
>>> print(output)
[2]