add type check

Author: yeliang2258

python/paddle/fluid/tests/unittests/test_compare_op.py

@@ -140,6 +140,17 @@ def test_dynamic_api_float(self):
                 self.assertEqual((out.numpy() == self.real_result).all(), True)
                 paddle.enable_static()
 
+        def test_assert(self):
+            def test_dynamic_api_string(self):
+                if self.op_type == "equal":
+                    paddle.disable_static()
+                    x = paddle.to_tensor(self.input_x)
+                    op = eval("paddle.%s" % (self.op_type))
+                    out = op(x, "1.0")
+                    paddle.enable_static()
+
+            self.assertRaises(TypeError, test_dynamic_api_string)
+
         def test_dynamic_api_bool(self):
             if self.op_type == "equal":
                 paddle.disable_static()

python/paddle/tensor/logic.py

@@ -175,6 +175,10 @@ def equal(x, y, name=None):
           result1 = paddle.equal(x, y)
           print(result1)  # result1 = [True False False]
     """
+    if not isinstance(y, (int, bool, float, Variable)):
+        raise TypeError(
+            "Type of input args must be float, bool, int or Tensor, but received type {}".
+            format(type(y)))
     if not isinstance(y, Variable):
         y = full(shape=[1], dtype=x.dtype, fill_value=y)