分组矩阵乘法的计算方法、装置、电子设备和存储介质

本发明涉及人工智能技术领域，提供一种分组矩阵乘法的计算方法、装置、电子设备和存储介质，其中方法包括：确定用于切分矩阵的切片形状；基于切片形状，对分组矩阵乘法中每个矩阵乘法的输入矩阵和输出矩阵进行切分，得到每个矩阵乘法的输入矩阵切片和输出矩阵切片；对所有矩阵乘法的输出矩阵切片进行集中编码，得到每个矩阵乘法中各输出矩阵切片的索引；基于各线程组和各输出矩阵切片的索引，对各输出矩阵切片对应的输入矩阵切片进行调度并执行矩阵乘法计算，得到各输出矩阵切片对应的计算结果。本发明通过确定统一的切片形状，并采取集中编码和调度的方式，可以提高处理问题的灵活性，并提高计算性能。