摘要
本发明公开了一种卷积权重梯度计算方法、装置、介质、设备及产品,所述方法包括:对全局内存中的输入特征图和对应的输出梯度图进行角色分配,确定动态滑动图和静态固定图;从动态滑动图和静态固定图中分别提取出滑动部分和对应的固定部分,并让每个线程持有滑动部分中一个数据行和对应的对齐数据块;在每个线程中,对持有的数据行和对齐数据块进行同步滑动,以及点乘运算,得到当前滑动位置的局部计算结果,并累加至权重梯度矩阵的缓存位置中,当所有滑动部分以及对应的固定部分同步遍历计算完成后,获得卷积核的权重梯度矩阵。本发明能够有效地实现滑动方向的数据复用,减少重复读取的数据量,从而提高权重梯度的计算效率。