功能描述
算子功能
用于计算稀疏矩阵与稠密矩阵的乘积。算子基于压缩稀疏行(CSR)存储结构设计,通过在装载与计算阶段跳过零块,实现计算与访存的高效利用。核心计算内核针对鲲鹏平台进行了SIMD优化。
算子公式
SparseGemm算子用于计算如下形式的矩阵乘法:

其中:
- A:稀疏矩阵
- B:稠密矩阵
- C:输出矩阵
- α,β:可选缩放系数
父主题: SparseGemm
用于计算稀疏矩阵与稠密矩阵的乘积。算子基于压缩稀疏行(CSR)存储结构设计,通过在装载与计算阶段跳过零块,实现计算与访存的高效利用。核心计算内核针对鲲鹏平台进行了SIMD优化。
SparseGemm算子用于计算如下形式的矩阵乘法:

其中: