
卷积计算加速
文章平均质量分 94
Matrix_1127
这个作者很懒,什么都没留下…
展开
-
卷积计算加速方法--slice卷积
slice卷积,每层分块的时候左边加上一些overlap,然后将卷积之后的结果slice一部分给右边的一块,保证他们的结果合并后和普通卷积结果一致,并且每层之间互不影响。原创 2023-06-21 15:57:50 · 1259 阅读 · 0 评论 -
卷积计算加速方法--分块卷积
当卷积的参数量太大导致内存不够用时,考虑将一大块卷积分成多个小块分别进行卷积,相当于将原始输入分成几个小的输入经过同一组卷积核分别卷积,其中每块小的输入都是原始输入的子集,每块之间互不影响,最后将结果合并,实现分块卷积的输出结果与整个输入卷积后的结果完全一致,这种分块卷积的算法可以减小内存消耗同时大大提高运行效率。因此在分块的时候会有overlap的出现,并且这个overlap会随着层数的增加会累积,如下这个函数就是用来计算输入每块overlap的尺寸。原创 2023-06-19 13:57:44 · 3052 阅读 · 0 评论