在python中并行化向量式计算的乘法

发布时间：2020-12-16 22:04:57 所属栏目：Python 来源：网络整理

导读：我有一大堆代码 for i in range(0,len(a)) b[i] = func(a[i]) 其中a和b是相同长度的数组,a是给定的(和大),func是一些具有大量局部变量但不使用任何全局变量的函数. 我想在几个CPU上分配func的计算.大概我需要使用多处理模块,但我没有找到任何相关的例子.你能

我有一大堆代码

for i in range(0,len(a))
    b[i] = func(a[i])

其中a和b是相同长度的数组,a是给定的(和大),func是一些具有大量局部变量但不使用任何全局变量的函数.

我想在几个CPU上分配func的计算.大概我需要使用多处理模块,但我没有找到任何相关的例子.你能帮忙吗？谢谢.

最佳答案

请参阅multiprocessing docs中的第一个代码示例：

from multiprocessing import Pool

# you could define `func`,`a` here

if __name__=="__main__":
    p = Pool() # use all available CPU cores
    b = p.map(func,a)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!