From 38cca30a47d9b557b725319a2c049cc028cc6522 Mon Sep 17 00:00:00 2001 From: Edgar Solomonik Date: Sun, 8 Dec 2019 15:25:06 -0600 Subject: [PATCH] correct W->V --- ccd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ccd.py b/ccd.py index 9f07693..3471522 100644 --- a/ccd.py +++ b/ccd.py @@ -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])