码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch

时间:2019-04-02 10:52:37      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:font   函数   不同的   style   spl   ann   问题   输入   参数   

Pytorch中的split问题:

1、使用torch.nn.Conv2d中有个参数是groups会将输入的feature map分组,此处需要注意的一点是分组之后各组的feature map的channel数目是相同的。

2、另外一种方式用torch.split函数将输入的feature map先split,之后再卷积出不同输出channels数目不同的feature maps。

pytorch

标签:font   函数   不同的   style   spl   ann   问题   输入   参数   

原文地址:https://www.cnblogs.com/yanxingang/p/10640910.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!