fix dtype check

PaddlePaddle · Aug 15, 2023 · f5889e1 · f5889e1
1 parent a067922
commit f5889e1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/tensor/logic.py b/python/paddle/tensor/logic.py
@@ -51,6 +51,8 @@ def _logical_op(op_name, x, y, out=None, name=None, binary_op=True):
                 "float32",
                 "float64",
                 "uint16",
+                "complex64",
+                "complex128",
             ],
             op_name,
         )
@@ -68,6 +70,8 @@ def _logical_op(op_name, x, y, out=None, name=None, binary_op=True):
                     "float32",
                     "float64",
                     "uint16",
+                    "complex64",
+                    "complex128",
                 ],
                 op_name,
             )