add multiplication by constant for gradU sym case

dahong67 · Nov 26, 2024 · 0739a38 · 0739a38
1 parent 3509af3
commit 0739a38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gcp-losses.jl b/src/gcp-losses.jl
@@ -118,6 +118,7 @@ function grad_U!(
     for j in 1:K
         if sym_data
             mttkrp!(GU[j], Y, tuple([M.U[k] for k in M.S]...), findall(M.S .== j)[1])
+            rmul!(GU[j], count(M.S .== j))
         else
             for (index, mode) in enumerate(findall(M.S .== j))
                 if index == 1