!1571 Equalcount check input shape same

Merge pull request !1571 from VectorSL/equalcount

mindspore/ops/operations/math_ops.py

@@ -1415,7 +1415,7 @@ class EqualCount(PrimitiveWithInfer):
     """
     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:
         - **input_x** (Tensor) - The first input tensor.
@@ -1438,6 +1438,7 @@ class EqualCount(PrimitiveWithInfer):
         self.init_prim_io_names(inputs=['x', 'y'], outputs=['output'])
 
     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,)
         return output_shape