[发明专利]运算装置和方法

说明书

本公开提供了一种运算装置，包括：运算控制模块，用于确定分块信息；运算模块，用于根据所述分块信息对运算矩阵进行分块、转置及合并运算，以得到所述运算矩阵的转置矩阵。本公开还提供了一种运算方法。本公开运算装置及方法，减小了存储资源和计算资源的开销，提高了运算速度。

本公开是申请号为201880001242.9中国专利的分案申请，母案专利的内容皆引用于此。

技术领域

本公开涉及人工智能技术领域，更具体地涉及一种运算装置和方法。

背景技术

多层神经网络被广泛应用于分类识别等任务，近年来，由于其较高的识别率和较高的可并行性，受到学术界和工业界的广泛关注。

目前一些性能较好的神经网络通常都非常庞大，这也意味着这些神经网络需要大量的计算资源和存储资源。大量的计算和存储资源的开销会降低神经网络的运算速度，同时，对硬件的传输带宽以及运算器的要求也大大提高。

发明内容

(一)要解决的技术问题

本公开提供了一种运算装置和方法，以至少部分解决以上所提出的技术问题。

(二)技术方案

根据本公开的一个方面，提供了一种运算装置，包括：

运算控制模块，用于确定分块信息；

运算模块，用于根据所述分块信息对运算矩阵进行分块、转置及合并运算，以得到所述运算矩阵的转置矩阵。

在一些实施例中，所述的运算装置，还包括：

地址存储模块，用于存储所述运算矩阵的地址信息；以及

数据存储模块，用于存储所述运算矩阵，并存储运算后的转置矩阵；

其中，所述运算控制模块用于从所述地址存储模块提取所述运算矩阵的地址信息，并根据所述运算矩阵的地址信息分析得到分块信息；所述运算模块，用于从所述运算控制模块获取运算矩阵的地址信息及分块信息，根据所述运算矩阵的地址信息从所述数据存储模块提取运算矩阵，并根据所述分块信息对所述运算矩阵进行分块、转置及合并运算，得到所述运算矩阵的转置矩阵，并将所述运算矩阵的转置矩阵反馈至所述数据存储模块。

在一些实施例中，所述运算模块包括矩阵分块单元、矩阵运算单元和矩阵合并单元，其中：

矩阵分块单元：用于从所述运算控制模块获取运算矩阵的地址信息及分块信息，并根据所述运算矩阵的地址信息从所述数据存储模块提取运算矩阵，根据所述分块信息对所述运算矩阵进行分块，得到n个分块矩阵；

矩阵运算单元，用于获取所述n个分块矩阵，并对所述n个分块矩阵进行转置运算，得到所述n个分块矩阵的转置矩阵；

矩阵合并单元，用于获取并合并所述n个分块矩阵的转置矩阵，得到所述运算矩阵的转置矩阵，并将所述运算矩阵的转置矩阵反馈至所述数据存储模块，其中，n为自然数。

在一些实施例中，所述运算模块还包括缓存单元，用于缓存所述n个分块矩阵，以供所述矩阵运算单元获取。

在一些实施例中，所述运算控制模块包括指令处理单元、指令缓存单元和矩阵判断单元，其中：

指令缓存单元，用于存储待执行的矩阵运算指令；

指令处理单元，用于从指令缓存单元中获取矩阵运算指令，对所述矩阵运算指令进行译码，并根据所述译码后的矩阵运算指令从所述地址存储模块中获取运算矩阵的地址信息；

矩阵判断单元，用于对所述运算矩阵的地址信息进行分析，得到所述分块信息。