Added code-src-note

2022-07-01 11:52:50 +02:00 · 2022-07-01 11:52:50 +02:00 · 2626ec82a6
commit 2626ec82a6
parent ab1b269af9
1 changed files with 4 additions and 0 deletions
--- a/metastable_baselines/misc/norm.py
+++ b/metastable_baselines/misc/norm.py
@ -3,6 +3,10 @@ from torch.distributions.multivariate_normal import _batch_mahalanobis


 def mahalanobis_alt(u, v, std):
+    """
+    Stolen from Fabian's Code (Public Version)
+
+    """
    delta = u - v
    return th.triangular_solve(delta, std, upper=False)[0].pow(2).sum([-2, -1])