c - 将 BLAS 中的三个矩阵相乘，中间一个是对角线

Question

A是一个MxK矩阵，B是一个大小为 的向量K，C是一个KxN矩阵。我应该使用哪组 BLAS 运算符来计算下面的矩阵？

M = A*diag(B)*C

实现这一点的一种方法是使用三个 for 循环，如下所示

for (int i=0; i<M; ++i)
    for (int j=0; j<N; ++j)
        for (int k=0; k<K; ++k)
            M(i,j) = A(i,k)*B(k)*C(k,j);

为了获得更好的速度效率，是否真的值得在 BLAS 中实现这一点？

Answer 1

首先计算D = diag(B)*C，然后使用适当的 BLAS 矩阵乘法来计算A*D。

您可以diag(B)*C使用对元素的循环B并调用适当的 BLAS 标量乘法例程来实现。