dmat_p can be done in place

JuliaStats · Jan 30, 2019 · 8e156b5 · 8e156b5
1 parent c5aa9d0
commit 8e156b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/kmeans.jl b/src/kmeans.jl
@@ -153,8 +153,8 @@ function _kmeans!(X::AbstractMatrix{T},
         else
             # if only a small subset is affected, only compute for that subset
             affected_inds = findall(to_update)
-            dmat_p = pairwise(distance, view(centers, :, affected_inds), X)
-            dmat[affected_inds, :] .= dmat_p
+            pairwise!(view(dmat, affected_inds, :), distance,
+                      view(centers, :, affected_inds), X)
         end
 
         # update assignments