fix AffineTransform.sign

randolf-scholz · randolf-scholz · commit 42411588073d · 2025-01-08T17:01:31.000+01:00
diff --git a/torch/distributions/transforms.py b/torch/distributions/transforms.py
@@ -776,7 +776,7 @@ def __eq__(self, other):
     def sign(self) -> int:
         if isinstance(self.scale, numbers.Real):
             return 1 if float(self.scale) > 0 else -1 if float(self.scale) < 0 else 0
-        return self.scale.sign()
+        return int(self.scale.sign().prod().item())  # type: ignore[union-attr]
 
     def _call(self, x):
         return self.loc + self.scale * x
