通用矩阵乘运算装置、方法、电子设备

本公开涉及计算机技术领域，提出一种通用矩阵乘运算装置、方法、电子设备。所述装置中，第一存储模块存储第一数据矩阵和第二数据矩阵，第一数据矩阵列数为s，第二数据矩阵行数为s；掩码生成模块根据第一数据矩阵和第二数据矩阵生成复合掩码；向量生成模块根据复合掩码和第一数据矩阵、第二数据矩阵生成第一向量和第二向量，第一向量的元素个数等于第二向量的元素个数且小于s，运算模块计算第一向量和第二向量的乘积，该乘积等于第一数据矩阵的第i行元素和第二数据矩阵的第j列元素的乘积；第二存储模块存储第一向量和第二向量的乘积。该装置实现的GEMM算法可降低要计算的数据量，从而降低装置的计算功耗、提升装置的能效。