Update removed torch.symeig in parametrizations.py (#2472)

kit1980 · lezcano · web-flow · commit 1e5ec44a131d · 2023-06-15T14:10:38.000-07:00
* Update removed torch.symeig from parametrizations.py torch.symeig was removed pytorch/pytorch#70988 --------- Co-authored-by: Mario Lezcano Casado <3291265+lezcano@users.noreply.github.com>
diff --git a/intermediate_source/parametrizations.py b/intermediate_source/parametrizations.py
@@ -255,7 +255,7 @@ def forward(self, X):
 parametrize.register_parametrization(layer_spd, "weight", MatrixExponential())
 X = layer_spd.weight
 print(torch.dist(X, X.T))                        # X is symmetric
-print((torch.symeig(X).eigenvalues > 0.).all())  # X is positive definite
+print((torch.linalg.eigvalsh(X) > 0.).all())  # X is positive definite
 
 ###############################################################################
 # Initializing parametrizations
