diff --git a/GPy/kern/_src/coregionalize.py b/GPy/kern/_src/coregionalize.py index cbfe1ccb..291402ec 100644 --- a/GPy/kern/_src/coregionalize.py +++ b/GPy/kern/_src/coregionalize.py @@ -154,9 +154,9 @@ class Coregionalize(Kern): def _gradient_reduce_numpy(self, dL_dK, index, index2): index, index2 = index[:,0], index2[:,0] dL_dK_small = np.zeros_like(self.B) - for i in range(k.output_dim): + for i in range(self.output_dim): tmp1 = dL_dK[index==i] - for j in range(k.output_dim): + for j in range(self.output_dim): dL_dK_small[j,i] = tmp1[:,index2==j].sum() return dL_dK_small