Fix eq none bug

python/oneflow/framework/tensor.py

@@ -119,7 +119,12 @@ def _meta_repr(self):
 
 
 def _eq(self, other):
-    return flow._C.equal(self, other)
+    if (self is None and other is None):
+        return True
+    elif (self is None and other is not None) or (self is not None and other is None):
+        return False
+    else:
+        return flow._C.equal(self, other)
 
 
 def _ne(self, other):

python/oneflow/test/tensor/test_tensor_part_1.py

@@ -1008,6 +1008,15 @@ def test_tensor_int(test_case):
         x = flow.tensor(2.3)
         y = int(x)
         test_case.assertTrue(np.array_equal(y, 2))
+
+    def test_none_equal(test_case):
+        xt = flow.randn(10)
+        yt = flow.randn(10)
+        z = None in [xt, yt]
+        test_case.assertTrue(np.array_equal(z, False))
+        zt = None
+        z = None in [xt, yt, zt]
+        test_case.assertTrue(np.array_equal(z, True))
 
 
 if __name__ == "__main__":