Skip to content

Commit

Permalink
correct W->V
Browse files Browse the repository at this point in the history
  • Loading branch information
solomonik committed Dec 8, 2019
1 parent e862f7c commit 38cca30
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ccd.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def run_CCD(T,U,V,W,omega,regParam,num_iter,time_limit,objective_frequency,use_M
if use_MTTKRP:
alphas = ctf.tensor(R.shape[2])
#ctf.einsum('ijk -> k', ctf.TTTP(R, [U_vec_list[f], V_vec_list[f], None]),out=alphas)
ctf.MTTKRP(R, [U_vec_list[f], W_vec_list[f], alphas], 2)
ctf.MTTKRP(R, [U_vec_list[f], V_vec_list[f], alphas], 2)
else:
alphas = ctf.einsum('ijk, i, j -> k', R, U_vec_list[f], V_vec_list[f])

Expand Down

0 comments on commit 38cca30

Please sign in to comment.