python - 获得两个大数组的线性组合的快速方法是什么？

Question

我有两个尺寸为 (87256 X 87256) 的 numpy 数组 A 和 B。

我有一个不断变化的单一因子 alpha。

我正在尝试找到最快的获取方式:

α * A + (1 - α) * B

使用简单的 numpy，计算大约需要 4 分 8 秒。

计算这个数组最快的方法是什么？我有一个多核系统，因此并行计算不是限制。

Answer 1

试试这两个选项:

选项-A

# multiplying only once
C = B + alpha*(A-B)

选项-B

我扩展了@Massifox 的回答。

import numba
import numpy as np

def get_result(A, B, alpha = 0.5):
    return B + alpha*(A-B)

get_result_jit = numba.vectorize(['float64(float64, float64, float64)'])(get_result)

其他资源:

1. Numba tutorial
2. 参见装饰器:@numba.vectorize , @numba.guvectorize和 @numba.jit and @numba.njit .
3. Speeding up algorithms with Numba