英特尔 MKL 和行乘法

Question

我有一个矩阵 S(n x m) 和一个向量 Sigma(n)，我想将每一行 S(i) 乘以 Sigma(i)。

我想到了三件事： -> 将 Sigma 转换为方形对角矩阵并计算 S = Sigma * S，但似乎这些函数仅适用于一般矩阵或三角矩阵...... -> 在循环中使用 DSCAL 将每行乘以标量 Sigma[i] -> mkl_ddiamm，但对我来说似乎有点晦涩。

关于我应该如何实施的任何建议？谢谢！

Answer 1

MKL/BLAS 没有为它提供函数是一个非常简单的操作。您可以使用 for 循环自己实现它。

for(int i=0; i<nrow; ++i) {
  for(int j=0; j<ncols; ++j) {
    s[i][j] += sigma[i];
  }
}