Added missing sqrt and simplified notation.

apache · Dec 17, 2023 · daf25e2 · daf25e2
1 parent 11e11b4
commit daf25e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/builtin/dist.dml b/scripts/builtin/dist.dml
@@ -33,7 +33,7 @@
 
 m_dist = function(Matrix[Double] X) return (Matrix[Double] Y) {
   n = nrow(X)
-  s = rowSums(X * X)
-  Y = - 2*X %*% t(X) + s + t(s)
+  s = rowSums(X^2)
+  Y = sqrt(-2 * X %*% t(X) + s + t(s))
   Y = replace(target = Y, pattern=NaN, replacement = 0);
 }