企业社会责任矩阵 - 矩阵乘法

CSR Matrix - Matrix multiplication

本文关键字：企业 社会责任      更新时间：2023-10-16

我有两个平方矩阵A和B

我必须将B转换为CSR Format并确定产品C 

A * B_csr = C

我在网上找到了很多关于CSR矩阵-向量乘法的信息。算法为：

for (k = 0; k < N; k = k + 1)
  result[i] = 0;
for (i = 0; i < N; i = i + 1)
{  
  for (k = RowPtr[i]; k < RowPtr[i+1]; k = k + 1)
  {  
    result[i] = result[i] + Val[k]*d[Col[k]];
  }  
}

但是，我需要Matrix - Matrix乘法。

此外，似乎大多数算法都适用于A_csr - vector我需要A * B_csr的乘法。我的解决方案是在转换之前转置两个矩阵，然后转置最终产品。

有人可以解释如何计算Matrix - CSR Matrix产品和/或CSR Matrix - Matrix产品吗？

这是 Python 中用于Dense Matrix X CSR Matrix的简单解决方案。它应该是不言自明的。

def main():
  # 4 x 4 csr matrix
  #    [1, 0, 0, 0],
  #    [2, 0, 3, 0],
  #    [0, 0, 0, 0],
  #    [0, 4, 0, 0],
  csr_values = [1, 2, 3, 4]
  col_idx    = [0, 0, 2, 1]
  row_ptr    = [0, 1, 3, 3, 4]
  csr_matrix = [
      csr_values,
      col_idx,
      row_ptr
      ]
  dense_matrix = [
      [1, 3, 3, 4],
      [1, 2, 3, 4],
      [1, 4, 3, 4],
      [1, 2, 3, 5],
      ]
  res = [
      [0, 0, 0, 0],
      [0, 0, 0, 0],
      [0, 0, 0, 0],
      [0, 0, 0, 0],
      ]
  # matrix order, assumes both matrices are square
  n = len(dense_matrix)
  # res = dense X csr
  csr_row = 0 # Current row in CSR matrix
  for i in range(n):
    start, end = row_ptr[i], row_ptr[i + 1]
    for j in range(start, end):
      col, csr_value = col_idx[j], csr_values[j]
      for k in range(n):
        dense_value = dense_matrix[k][csr_row]
        res[k][col] += csr_value * dense_value
    csr_row += 1
  print res

if __name__ == '__main__':
  main()

CSR Matrix X Dense Matrix真的只是密集矩阵的每一行CSR Matrix X Vector乘积序列，对吗？因此，扩展上面显示的代码来执行此操作应该非常容易。

展望未来，我建议你不要自己编写这些例程。如果您使用的是C++（基于标签），那么您可以看看例如Boost ublas或Eigen。这些 API 乍一看似乎有点晦涩难懂，但从长远来看，这确实是值得的。首先，您可以访问更多功能，您将来可能需要这些功能。其次，这些实现将得到更好的优化。

相关文章：