fix typo of torch.eig (#281)

Co-authored-by: glynpu <glynwpu@qq.com>
This commit is contained in:
LIyong.Guo 2022-03-31 10:43:46 +08:00 committed by GitHub
parent 2045125fd9
commit fc40bfea82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -140,7 +140,7 @@ def get_diagnostics_for_dim(
stats = eigs.abs().sqrt() stats = eigs.abs().sqrt()
except: # noqa except: # noqa
print("Error getting eigenvalues, trying another method.") print("Error getting eigenvalues, trying another method.")
eigs, _ = torch.eigs(stats) eigs = torch.linalg.eigvals(stats)
stats = eigs.abs().sqrt() stats = eigs.abs().sqrt()
# sqrt so it reflects data magnitude, like stddev- not variance # sqrt so it reflects data magnitude, like stddev- not variance
elif sizes_same: elif sizes_same: