pytorch
Posted yanxingang
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch相关的知识，希望对你有一定的参考价值。
Pytorch中的split问题：
1、使用torch.nn.Conv2d中有个参数是groups会将输入的feature map分组，此处需要注意的一点是分组之后各组的feature map的channel数目是相同的。
2、另外一种方式用torch.split函数将输入的feature map先split，之后再卷积出不同输出channels数目不同的feature maps。
以上是关于pytorch的主要内容，如果未能解决你的问题，请参考以下文章