摘要
本发明公开了一种卷积权重梯度计算方法、装置、介质、设备及产品,所述方法包括:对全局内存中的输入特征图和对应的输出梯度图进行角色分配,确定动态滑动图和静态固定图;从动态滑动图和静态固定图中分别提取出滑动部分和对应的固定部分,并让每个线程持有滑动部分中一个数据行和对应的对齐数据块;在每个线程中,对持有的数据行和对齐数据块进行同步滑动,以及点乘运算,得到当前滑动位置的局部计算结果,并累加至权重梯度矩阵的缓存位置中,当所有滑动部分以及对应的固定部分同步遍历计算完成后,获得卷积核的权重梯度矩阵。本发明能够有效地实现滑动方向的数据复用,减少重复读取的数据量,从而提高权重梯度的计算效率。
技术关键词
计算方法
人工智能处理器
数据
动态
可读存储介质
矩阵
内存
通道
计算机程序产品
模块
尺寸
存储器
元素
电子设备
批量
机制
资源
系统为您推荐了相关专利信息
天气预测模型
协调优化方法
风电设备
光电设备
光伏储能
功率因数
多元线性回归模型
数据
专用变压器
处理器