Fix pytorch warning when apply logical_xor on different shape tensor.

USTC-TNS · Nov 22, 2023 · 901298d · 901298d
1 parent c071053
commit 901298d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tat/_utility.py b/tat/_utility.py
@@ -21,9 +21,9 @@ def neg_symmetry(tensor: torch.Tensor) -> torch.Tensor:
 
 def add_symmetry(tensor_1: torch.Tensor, tensor_2: torch.Tensor) -> torch.Tensor:
     if tensor_1.dtype is torch.bool:
-        return torch.logical_xor(tensor_1, tensor_2)
+        return tensor_1 ^ tensor_2
     else:
-        return torch.add(tensor_1, tensor_2)
+        return tensor_1 + tensor_2
 
 
 def zero_symmetry(tensor: torch.Tensor) -> torch.Tensor: