如何使进程能够在主程序的数组中编写？

Question

我正在创建一个进程池，并且每个进程都需要写在主程序中存在的矩阵的不同部分中。不必担心会覆盖信息，因为每个进程将与矩阵的不同行一起工作。我如何使矩阵在过程中可写？

该程序是一位教授分配给我的矩阵乘法器，必须进行多次处理。它将为计算机具有的每个核心创建一个进程。主程序将矩阵的不同部分发送给流程，它们将对它们进行计算，然后以一种我可以识别出哪个响应对应于它所基于的行的方式将它们返回。

Answer 1

矩阵乘法是指分别计算所得矩阵的每个元素。对于Pool来说，这似乎是一份工作。由于这是家庭作业（还要遵循SO代码），因此我仅说明Pool本身的用法，而不是整个解决方案。

因此，您必须编写一个例程来计算所得矩阵的第（i，j）个元素：

def getProductElement(m1, m2, i, j):
    # some calculations
    return element

然后初始化池：

from multiprocessing import Pool, cpu_count
pool = Pool(processes=cpu_count())

然后，您需要提交工作。您也可以将它们组织成矩阵，但是为什么麻烦，让我们列出一个清单。

result = []
# here you need to iterate through the the columns of the first and the rows of
# the second matrix. How you do it, depends on the implementation (how you store
# the matrices). Also, make sure you check the dimensions are the same.
# The simplest case is if you have a list of columns:

N = len(m1)
M = len(m2[0])
for i in range(N):
    for j in range(M):
        results.append(pool.apply_async(getProductElement, (m1, m2, i, j)))

然后用结果填充结果矩阵：

m = []
count = 0
for i in range(N):
    column = []
    for j in range(M):
        column.append(results[count].get())
    m.append(column)

同样，代码的确切形状取决于您如何表示矩阵。