[发明专利]一种多输入多输出矩阵卷积的并行处理方法有效
| 申请号： | 201711473269.1 | 申请日： | 2017-12-29 |
| 公开（公告）号： | CN108205702B | 公开（公告）日： | 2020-12-01 |
| 发明（设计）人： | 郭阳;张军阳;杨超;田希;扈啸;李斌 | 申请（专利权）人： | 中国人民解放军国防科技大学 |
| 主分类号： | G06N3/04 | 分类号： | G06N3/04;G06N3/08 |
| 代理公司： | 湖南兆弘专利事务所(普通合伙) 43008 | 代理人： | 周长清 |
| 地址： | 410073 湖南*** | 国省代码： | 湖南;43 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 一种多输入多输出矩阵卷积的并行处理方法，其步骤为：S1：依据向量处理器的向量处理单元VPE的数量N等参数确定输出特征图的最优计算方案；S2：将M张输入特征图依次存入外部存储DDR中，将N个输入卷积核按照第3维，按行的方式进行拼接；S3：加载输入特征图1的第一个元素，广播至向量寄存器，同时加载AM中卷积核的第一行元素至向量寄存器中；S4：进行k×k次累加，完成输入特征图1的计算；同时加载第2幅输入特征图；S5：重复直至完成N个输出特征图第一个元素的计算；S6：根据移动步长完成N个输出特征图所有元素的计算；S7：以上步骤循环最终完成。本发明具有实现简单、操作方便、可提高向量处理器并行性、能提高处理器运算效率等优点。 | ||
| 搜索关键词： | 一种 输入 输出 矩阵 卷积 并行 处理 方法 | ||
【主权项】：
1.一种多输入多输出矩阵卷积的并行处理方法，其特征在于，步骤为：S1：依据向量处理器的向量处理单元VPE的数量N、输入特征图的数量M，卷积核的数量P，卷积核的尺寸k，移动步长s，确定输出特征图的最优计算方案；S2：将M张输入特征图依次存入外部存储DDR中，将N个输入卷积核按照第3维，按行的方式进行拼接，并将完成拼接后的卷积核矩阵传入向量处理器的向量存储体中；其中，N<＝p；S3：加载输入特征图1的第一个元素，广播至向量寄存器，同时加载AM中卷积核的第一行元素至向量寄存器中，进行乘加并累加至累加寄存器中；S4：进行k×k次累加，完成输入特征图1的计算；同时加载第2幅输入特征图，并加载第2个卷积核矩阵的第一行元素；S5：重复S3‑S4直至完成N个输出特征图第一个元素的计算；S6：根据水平或垂直方向的移动步长，移动s个元素的距离，重复步骤S3‑S5直至完成N个输出特征图所有元素的计算；S7：以上步骤共循环P/N次，最终完成p个输出特征图所有元素的计算。
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于中国人民解放军国防科技大学，未经中国人民解放军国防科技大学许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/201711473269.1/，转载请声明来源钻瓜专利网。
