Polish code

23291abd · minqiyang · 7c767a44 · 23291abd
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +2 -1

未找到文件。
--- 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: