多维张量的规约计算方法、装置、设备及介质

摘要

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