我使用 NumPy 函数einsum https://docs.scipy.org/doc/numpy/reference/generated/numpy.einsum.html执行两个 2D NumPy 数组的逐元素乘法和求和。
np.einsum('ij,ij',A,B)
每个数组 A、B 的大小都围绕10000 x 10000
。我注意到此操作是我的代码中的瓶颈,占用了大约 85% 的处理时间。如何快速并行化此操作?
None
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)