Feat (core): simplify binary_sign impl (#672)

Xilinx · Jul 17, 2023 · 2622a2b · 2622a2b
1 parent 32fba9e
commit 2622a2b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/brevitas/function/ops.py b/src/brevitas/function/ops.py
@@ -28,8 +28,7 @@ def binary_sign(x: Tensor) -> Tensor:
         tensor([ 1., -1.,  1.])
     """
     positive_mask = torch.ge(x, 0.0)
-    negative_mask = torch.lt(x, 0.0)
-    y = positive_mask.to(x.dtype) - negative_mask.to(x.dtype)
+    y = positive_mask.to(x.dtype) * 2. - 1.
     return y