一种用于稀疏矩阵向量乘设计的神经网络模型

本发明提供了一种用于稀疏矩阵向量乘设计的神经网络模型，用于为稀疏矩阵向量乘程序设计相关的多个决策阶段分别提供用于决策的策略和价值，所述神经网络包括：特征提取器，用于从稀疏矩阵在前一阶段的阶段表达中提取阶段特征，所述阶段表达包括层次映射、非零元的绝对位置和相对位置，所述层次映射指示稀疏矩阵的非零元在计算硬件中的分配信息；策略网络，用于根据所述阶段特征确定当前阶段的策略，所述策略为预测的当前阶段采取各动作的概率。价值网络，用于根据所述阶段特征确定价值，所述价值为前一阶段的阶段表达对应状态的期望价值。