Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1571 Equalcount check input shape same 

Merge pull request !1571 from VectorSL/equalcount
This commit is contained in:
mindspore-ci-bot 2020-05-28 15:26:43 +08:00 committed by Gitee
parent c1c359a709 4a8356b19a
commit 205cfec632
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/operations/math_ops.py
View File

@ -1415,7 +1415,7 @@ class EqualCount(PrimitiveWithInfer):
""" """
Computes the number of the same elements of two tensors. Computes the number of the same elements of two tensors.
The two input tensors should have same data type. The two input tensors should have same data type and shape.
Inputs: Inputs:
- **input_x** (Tensor) - The first input tensor. - **input_x** (Tensor) - The first input tensor.
@ -1438,6 +1438,7 @@ class EqualCount(PrimitiveWithInfer):
self.init_prim_io_names(inputs=['x', 'y'], outputs=['output']) self.init_prim_io_names(inputs=['x', 'y'], outputs=['output'])
def infer_shape(self, x_shape, y_shape): def infer_shape(self, x_shape, y_shape):
validator.check("x_shape", x_shape, "y_shape", y_shape, Rel.EQ, self.name)
output_shape = (1,) output_shape = (1,)
return output_shape return output_shape