Fix invert (~) operator for bool

tum-pbs · Nov 24, 2023 · 1bf1dea · 1bf1dea
1 parent f0f4086
commit 1bf1dea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/phiml/backend/_backend.py b/phiml/backend/_backend.py
@@ -1613,6 +1613,11 @@ def shift_bits_right(self, a, b):
         a, b = self.auto_cast(a, b)
         return a >> b
 
+    def invert(self, x):
+        if isinstance(x, bool):
+            return not x
+        return ~x
+
 
 BACKENDS: List[Backend] = []
 """ Global list of all registered backends. Register a `Backend` by adding it to the list. """

diff --git a/phiml/math/_tensors.py b/phiml/math/_tensors.py
@@ -714,7 +714,7 @@ def __neg__(self):
         return self._op1(lambda t: -t)
 
     def __invert__(self):
-        return self._op1(lambda t: ~t)
+        return self._op1(lambda t: choose_backend(t).invert(t))
 
     def __reversed__(self):
         assert self.shape.channel.rank == 1