[发明专利]矩阵乘法器

摘要

本发明实施例公开了一种矩阵转乘法器，涉及数据计算技术领域，旨在对两个矩阵进行分块计算。该矩阵乘法器包括：第一存储器、第二存储器、运算电路和控制器，其中，运算电路与第一存储器和第二存储器可以通过总线进行数据通信，控制器用于依据预设的程序或者指令控制第一矩阵和第二矩阵进行分块，并控制运算电路根据控制器的分块结果对第一存储器和第二存储器中的对应分块进行乘法运算。该矩阵乘法器可以用于对两个矩阵进行相乘运算。

主权项

1.一种矩阵乘法器，其特征在于，包括：第一存储器，用于存储第一矩阵，所述第一矩阵为M*K矩阵；第二存储器，用于存储第二矩阵，所述第二矩阵为K*N矩阵；与所述第一存储器和所述第二存储器连接的运算电路，所述运算电路包括X行*Y列运算单元，每个运算单元包括向量乘法电路和加法电路，所述矩阵乘法电路用于接收所述第一存储器发送的行向量数据和第二存储器发送的列向量数据，并将所述两路向量相乘；所述加法电路用于对所述两路向量相乘的结果相加，以及对属于同一个运算单元的计算结果进行累加得到每个运算单元的运算结果；与所述运算电路连接的控制器，所述控制器用于执行以下动作：将所述第一矩阵以规模为X*L的子块为单位进行分块，得到S×R个相同大小的子块，其中，将所述S×R个子块的第s行第r列的子块记为A_sr，s＝(1、2、3、……S)，r＝(1、2、3、……R)；将所述第二矩阵以规模为L*Y的子块为单位进行分块，得到R×T个相同大小的子块，其中，将所述R×T个子块中的第r行第t列记为B_rt，r＝(1、2、3、……R)，t＝(1、2、3、……T)；所述控制器还用于执行以下动作：将任意一个子块A_sr的X个行向量中的第x行和对应的子块B_rt的Y个列向量中的第y列，输入到所述X行*Y列运算单元中的第x行第y列的运算单元中进行运算，x＝(1、2、3、……X)，y＝(1、2、3、……Y)，其中，所述任意一个子块A_sr中的r和所述对应的子块B_rt中的r的取值相等。