这个屌丝很懒，什么也没留下！

pytorch底层组卷积的实现方式_pytorch底层是如何进行卷积计算的

作者：煮酒与君饮 | 2024-07-04 14:51:28

pytorch底层是如何进行卷积计算的

今天因为个人需求，去查看了一波pytorch源码中组卷积的实现，所以来记录一波。

源码见网页https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/Convolution.cpp#L289-L297

简单的说就是把通道split成group个数目，然后分别进行卷积，然后cat起来。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/煮酒与君饮/article/detail/787291