[发明专利]一种提高涉及稀疏矩阵计算速率的方法和系统

说明书

本说明书实施例公开了一种提高涉及稀疏矩阵计算速率的方法及系统。所述方法包括：获取稀疏矩阵，所述稀疏矩阵由至少一个非零值和所述至少一个非零值对应的坐标组成，所述坐标表示所述非零值在所述稀疏矩阵中的位置；所述坐标包括行坐标，所述行坐标表示所述非零值在所述稀疏矩阵中的行数；遍历所述稀疏矩阵，基于所述非零值对应的行坐标，生成所述至少一个非零值的至少一个访问索引；其中，所述访问索引表示所述非零值在所述至少一个非零值中被访问的顺序，行坐标相同的所述非零值对应的访问索引相邻；基于所述至少一个访问索引，访问所述至少一个非零值，并基于所述至少一个非零值进行矩阵计算。

技术领域

本说明书涉及数据处理技术领域，特别涉及一种访问稀疏矩阵的方法和系统。

背景技术

稀疏矩阵广泛的应用于各种机器学习模型中，例如：图深度学习的核心步骤-信息聚合，可以表示为一个稀疏矩阵和稠密矩阵的乘法。有些情况下，稀疏矩阵中的非零值没有按行的顺序存储，而为了优化计算过程需要读取稀疏矩阵中整行的非零值。为了可以按行的顺序来访问稀疏矩阵，常规的解决方案需要对稀疏矩阵中的非零值进行排序，对于大规模的稀疏矩阵，因为排序过程中涉及到大量的内存复制，效率较低。

因此，需要提供一种按行访问稀疏矩阵的方法和系统。

发明内容

本说明书实施例之一提供一种提高涉及稀疏矩阵计算速率的方法。所述方法包括：获取稀疏矩阵，所述稀疏矩阵由至少一个非零值和所述至少一个非零值对应的坐标组成，所述坐标表示所述非零值在所述稀疏矩阵中的位置；所述坐标包括行坐标，所述行坐标表示所述非零值在所述稀疏矩阵中的行数；遍历所述稀疏矩阵，基于所述非零值对应的行坐标，生成所述至少一个非零值的至少一个访问索引；其中，所述访问索引表示所述非零值在所述至少一个非零值中被访问的顺序，行坐标相同的所述非零值对应的访问索引相邻；基于所述至少一个访问索引，访问所述至少一个非零值，并基于所述至少一个非零值进行矩阵计算。

本说明书实施例之一提供一种提高涉及稀疏矩阵计算速率的系统，所述系统包括：获取模块，用于获取稀疏矩阵，所述稀疏矩阵由至少一个非零值和所述至少一个非零值对应的坐标组成，所述坐标表示所述非零值在所述稀疏矩阵中的位置；所述坐标包括行坐标，所述行坐标表示所述非零值在所述稀疏矩阵中的行数；生成模块，用于遍历所述稀疏矩阵，基于所述非零值对应的行坐标，生成所述至少一个非零值的至少一个访问索引；其中，所述访问索引表示所述非零值在所述至少一个非零值中被访问的顺序，行坐标相同的所述非零值对应的访问索引相邻；访问模块，用于基于所述至少一个访问索引，访问所述至少一个非零值，并基于所述至少一个非零值进行矩阵计算。

本说明书实施例之一提供一种提高涉及稀疏矩阵计算速率的装置，包括：

所述装置包括至少一个处理器以及至少一个存储器；所述至少一个存储器用于存储计算机指令；所述至少一个处理器用于执行所述计算机指令中的至少部分指令以实现一种提高涉及稀疏矩阵计算速率的方法。

附图说明

本说明书将以示例性实施例的方式进一步说明，这些示例性实施例将通过附图进行详细描述。这些实施例并非限制性的，在这些实施例中，相同的编号表示相同的结构，其中：

图1是根据本说明书一些实施例所示的提高涉及稀疏矩阵计算速率的系统的模块图；

图2是根据本说明书一些实施例所示的提高涉及稀疏矩阵计算速率的方法的示例性流程图；

图3A是根据本说明书一些实施例所示的稀疏矩阵的示例的示意图；

图3B是根据本说明书一些实施例所示的稀疏矩阵中非零值存储方式的示例的示意图；

图3C是根据本说明书一些实施例所示的按照访问索引存储非零值在非零值数组中位置的示例的示意图；

图3D是根据本说明书一些实施例所示的按行的顺序读取出来的非零值的示例的示意图；以及