mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a measure of correlation for eigs that can be negative.

Browse Source
This commit is contained in:
Daniel Povey 2022-07-26 13:40:57 +08:00
parent b9696878b4
commit e25ca74955
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
icefall/diagnostics.py
View File

@ -324,7 +324,11 @@ class TensorDiagnostic(object):
if stats.ndim == 2:
# Matrices, for purposes of measuring eigenvalues. Just compute a dot-product-related
# measure of correlation.
# measure of correlation of eigenvalues.
def norm_diag_mean(x):
return x - torch.eye(x.shape[0], dtype=x.dtype, device=x.device) * x.diag().mean()
stats = norm_diag_mean(stats)
other_stats = norm_diag_mean(other_stats)
correlation = ((stats * other_stats).sum() /
((stats**2).sum() * (other_stats**2).sum() + 1.0e-20).sqrt())
else: