[发明专利]多维张量的规约计算方法、装置、设备及介质

说明书

本发明实施例公开了一种多维张量的规约计算方法、装置、设备及介质。该方法包括：获取待规约计算的多维张量，以及多维张量的待规约维度信息和非待规约维度信息；根据待规约维度信息和非待规约维度信息，对多维张量的维度信息进行预处理，并根据预处理结果将多维张量调整为二维张量形式；其中，二维张量的高维度为待规约维度，低维度为非待规约维度；根据二维张量低维度信息，将二维张量划分为至少两块数据区域，并调用匹配数量的硬件计算单元分别对每块数据区域进行规约计算；将各个硬件计算单元得到的规约计算子区域向量进行顺序组合，得到与多维张量对应的规约计算结果。上述技术方案提高了对多维张量的规约计算速度。

技术领域

本发明实施例涉及智能计算技术领域，尤其涉及一种多维张量的规约计算方法、装置、设备及介质。

背景技术

随着人工智能以及高性能计算的蓬勃发展，人们对计算机算力的需求也成指数级增长，同时，算力的增长速度也极大地促进了人工智能以及高性能计算的发展。

在人工智能以及高性能计算领域，规约计算是一种常用的计算模式，规约计算的运算速度直接影响系统整体的算力输出，然而，传统的规约计算方法速度较慢，已经不能满足当前急速增长的算力需求。

发明内容

本发明实施例提供一种多维张量的规约计算方法、装置、设备及介质，以提高对多维张量的规约计算速度。

第一方面，本发明实施例提供了一种多维张量的规约计算方法，包括：

获取待规约计算的多维张量，以及与所述多维张量的对应的待规约维度信息和非待规约维度信息；

根据所述待规约维度信息和非待规约维度信息，对所述多维张量的维度信息进行预处理，并根据预处理结果将所述多维张量调整为二维张量形式；其中，所述二维张量的高维度为待规约维度，低维度为非待规约维度；

根据所述二维张量低维度信息，将所述二维张量划分为至少两块数据区域，并调用匹配数量的硬件计算单元分别对每块数据区域进行规约计算；

将各个所述硬件计算单元得到的规约计算子区域向量进行顺序组合，得到与所述多维张量对应的规约计算结果。

第二方面，本发明实施例还提供了一种多维张量的规约计算装置，包括：

张量信息获取模块，用于获取待规约计算的多维张量，以及与所述多维张量的对应的待规约维度信息和非待规约维度信息；

张量调整模块，用于根据所述待规约维度信息和非待规约维度信息，对所述多维张量的维度信息进行预处理，并根据预处理结果将所述多维张量调整为二维张量形式；其中，所述二维张量的高维度为待规约维度，低维度为非待规约维度；

规约计算模块，用于根据所述二维张量低维度信息，将所述二维张量划分为至少两块数据区域，并调用匹配数量的硬件计算单元分别对每块数据区域进行规约计算；

规约计算结果生成模块，用于将各个所述硬件计算单元得到的规约计算子区域向量进行顺序组合，得到与所述多维张量对应的规约计算结果。

第三方面，本发明实施例还提供了一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现本发明任意实施例所述的多维张量的规约计算方法。

第四方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现本发明任意实施例所述的多维张量的规约计算方法。