Modify computation outer tensor-vector products

dahong67 · Nov 28, 2023 · a933580 · a933580
1 parent f869de6
commit a933580
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/gcp-opt.jl b/src/gcp-opt.jl
@@ -199,8 +199,7 @@ function mttkrp(X, U, n)
         sz = size(Rn_j)
         m = length(sz)
         for k in n-1:-1:1
-            Rn_j = reshape(Rn_j, prod(sz[1:m-1]), sz[m])
-            Rn_j = Rn_j * U[k][:, j]
+            Rn_j = reshape(Rn_j, prod(sz[1:m-1]), sz[m]) * U[k][:, j]
             m -= 1
         end
         Rn[:, j] = Rn_j