[发明专利]一种矩阵数据处理方法、装置、设备及存储介质

说明书

本发明涉及矩阵处理领域，具体是涉及一种矩阵数据处理方法、装置、设备及存储介质。本发明将矩阵的元素加载到线程中，元素在线程中进行流动，使得两个矩阵中的相应元素能够进行乘积运算，且元素参与一次运算之后继续在线程中流动，以进行下次乘积运算。通过上述分析可知，即使两个矩阵进行乘法运算，各个元素需要多次参与计算，但由于本申请的元素是在线程中的，因此不需要往复从存储器中读取元素，所以本申请能够节省读取元素所需要的时间，进而使得本申请能够提高矩阵乘法的运行速度。

技术领域

本发明涉及矩阵处理技术领域，具体是涉及一种矩阵数据处理方法、装置、设备及存储介质。

背景技术

矩阵在进行乘法运行时，矩阵中的各个元素会多次参与计算，各个元素每参与一次计算就需要从存储器中读取一次各个元素，每次读取元素都需要一定的读取时间。一方面因为读取时间而增加了矩阵乘法运算所需要的运行时间；另一方面当读取时间与计算时间不匹配时(即已经完成上一次的计算，但是下一次计算所需要的元素还没有完成读取这一操作)，也会增加两个矩阵乘法运算所需要的运行时间。从上述分析可知从存储器中读取矩阵元素以进行矩阵乘法会降低矩阵乘法运算的速度。

综上所述，现有技术矩阵乘法的运算速度受到数据读取速度的限制。

因此，现有技术还有待改进和提高。

发明内容

为解决上述技术问题，本发明提供了一种矩阵数据处理方法、装置、设备及存储介质，提高了数据运算的吞吐率，加速了矩阵乘法运算。

为实现上述目的，本发明采用了以下技术方案：

第一方面，本发明提供一种矩阵数据处理方法，其中，包括：

获取第一待运算矩阵和第二待运算矩阵，所述第一待运算矩阵的列数等于所述第二待运算矩阵的行数；

通过线程按所述第二待运算矩阵的列读取所述第二待运算矩阵、读取所述第一待运算矩阵中与所述第二待运算矩阵的列相对应的匹配元素；

将所述线程中的所述第一待运算矩阵的元素与所述第二待运算矩阵的列中对应元素相乘，得到初始运算结果；

将所述匹配元素在所述线程之间以脉动的方式进行流动，计算每流动一次所述线程中的所述匹配元素与所述列元素相乘所对应的运算结果，直至所述第一待运算矩阵的所有元素流经过所有所述线程，并将每流动一次所述线程中的所述匹配元素与所述列元素相乘所对应的运算结果记为流动之后的运算结果；

通过所述线程中的所述初始运算结果和流动之后的各个所述运算结果，得到目标结果，所述目标结果用于反映所述第一待运算矩阵乘所述第二待运算矩阵所对应的目标矩阵。

在一种实现方式中，所述通过线程按所述第二待运算矩阵的列读取所述第二待运算矩阵、读取所述第一待运算矩阵中与所述第二待运算矩阵的列相对应的匹配元素，包括：

依据所述第一待运算矩阵，得到所述第一待运算矩阵中的行矩阵；

依据所述第二待运算矩阵，得到所述第二待运算矩阵中的二维矩阵；

通过各个所述线程按所述二维矩阵的各列读取所述二维矩阵；

通过各个所述线程读取所述行矩阵中与所述二维矩阵的各列所对应的各个行元素。

在一种实现方式中，所述得到所述初始运算结果之后，将所述匹配元素在所述线程之间以脉动的方式进行流动，计算每流动一次所述线程中的所述匹配元素与所述列元素相乘所对应的运算结果，直至所述第一待运算矩阵的所有元素流经过所有所述线程，并将每流动一次所述线程中的所述匹配元素与所述列元素相乘所对应的运算结果记为流动之后的运算结果，包括：