python - 使用 Numba 并行化矩阵计算

Question

我正在尝试在 Python 3.6 中并行化以下函数：

def get_Q(predictors, d):
    Q = 0.
    for i in range(len(predictors)):
        for j in range(i+1, len(predictors)):
            ci = predictors[i,:]
            cj = predictors[j,:]
            Q += product_cols(ci, cj, d[i], d[j])
    return Q

其中 :
predictors[,]是一个输入数组，具有n-rows 和p-features，并且
d[]是一个-sizen数组。

函数product_cols()定义为：

def product_cols(ci, cj, di, dj):
    c = ci-cj
    return np.matmul(np.transpose(np.matrix(c)), np.matrix(c))*((di-dj)**2)

其中 :
ci[]是 -大小cj[]的p数组并且
di, dj是实数。

有人可以帮我简单地并行化吗？