performance - 向量化乘以不同的常数

Question

我正在尝试将各种矩阵乘以常数，并且通常矢量化会使这样的事情更快，但我无法完全弄清楚这一点。假设我正在做一个多项式展开，并且我有一个用于不同函数的常量列表。

    for i in range(10):
        expansions[i,:,:] += c[i,k]*(matrix)

现在，这是做这件事的简单方法，但我想知道是否有一些方法可以做到这一点：

        expansions[:,:,:] += c[:,k]*(matrix)

我将每个常量单独应用于“矩阵”。我试图在自身之上堆叠许多“矩阵”，以创建大小（10xNxN）的东西，但这没有用。有任何想法吗？

标签： performancenumpyvectorization