Python中矩阵乘法的最快方法是什么？

Question

我正在 Python 中编写一个以速度为主要驱动程序的应用程序。 在优化我的代码时，我发现主要的瓶颈是用于计算的代码

在我的代码中，这个矩阵乘法计算为

POW = np.arange(4)
y = C @ (x ** POW)

我尝试使用不同的方法（例如for循环和其他方法），但现在这是我发现的最快的方法。 您对提高计算时间有什么建议吗？

Answer 1

绝对要使用 Numpy。 Numpy 在高度优化的 C 代码中执行实际的数学运算。 使用 Numpy 比编写自己的非优化 C 代码更快。

Answer 2

首先，使用 float 而不是 int。 其次，如果您不需要双精度，请使用 np.float32。

POW = np.arange(4, dtype='f')
# C = C.astype('f', copy=False) # ensure that C.dtype == np.float32
y = C @ (x ** POW)