Fix CayleyMap's right_inverse, s/X/A (#2765)

This makes it correct.

Co-authored-by: Svetlana Karslioglu <[email protected]>

Author: afrozenator

intermediate_source/parametrizations.py

@@ -306,8 +306,8 @@ def forward(self, X):
     def right_inverse(self, A):
         # Assume A orthogonal
         # See https://en.wikipedia.org/wiki/Cayley_transform#Matrix_map
-        # (X - I)(X + I)^{-1}
-        return torch.linalg.solve(X + self.Id, self.Id - X)
+        # (A - I)(A + I)^{-1}
+        return torch.linalg.solve(A + self.Id, self.Id - A)
 
 layer_orthogonal = nn.Linear(3, 3)
 parametrize.register_parametrization(layer_orthogonal, "weight", Skew())