Polish code

7 years ago · 23291abdb6
parent 7c767a44c2
commit 23291abdb6
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -6540,7 +6540,8 @@ for func in [
 def _logical_op(op_name, x, y, out=None, name=None, binary_op=True):
    helper = LayerHelper(op_name, **locals())

-    assert x.dtype == y.dtype
+    if binary_op:
+        assert x.dtype == y.dtype

    if out is None:
        if name is None: