基于数据流芯片进行批量矩阵乘计算的方法和装置

AITNT
正文
推荐专利
基于数据流芯片进行批量矩阵乘计算的方法和装置
申请号:CN202410951951
申请日期:2024-07-15
公开号:CN119025805A
公开日期:2024-11-26
类型:发明专利
摘要
本发明提出一种基于数据流芯片进行批量矩阵乘计算的方法和装置、电子设备以及非瞬时性计算机可读存储介质。根据本发明的实施例,所述方法包括响应于计算指令,分别对用于批量矩阵乘计算的第一输入维度数据和第二输入维度数据进行转置变换操作;利用第一输入维度数据或第二输入维度数据确定分组卷积组数;基于数据流芯片,利用经过转置变换操作的第一输入维度数据、经过转置变换操作的第二输入维度数据和分组卷积组数进行卷积计算;对卷积计算结果进行变换转置操作,得到批量矩阵乘计算的计算结果。根据实施例,通过对输入维度数据进行转置变换操作,将批量矩阵乘计算转换为分组卷积计算,从而利用数据流芯片实现批量矩阵乘计算,以提高计算效率。
技术关键词
非瞬时性计算机可读存储介质 数据 批量 矩阵 芯片 计算机可读指令 处理器 元素 电子设备 存储器
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号