diff --git a/intermediate_source/parametrizations.py b/intermediate_source/parametrizations.py index 086a4300674..096836e933b 100644 --- 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