pytorch中nn.Conv1d的用法详解


Posted in Python onDecember 31, 2019

先粘贴一段official guide:nn.conv1d官方

pytorch中nn.Conv1d的用法详解

我一开始被in_channels、out_channels卡住了很久,结果发现就和conv2d是一毛一样的。话不多说,先粘代码(菜鸡的自我修养)

class CNN1d(nn.Module):

  def __init__(self):
    super(CNN1d,self).__init__()
    self.layer1 = nn.Sequential(
          nn.Conv1d(1,100,2),
          nn.BatchNorm1d(100),
          nn.ReLU(),
          nn.MaxPool1d(8))
    self.layer2 = nn.Sequential(
          nn.Conv1d(100,50,2),
          nn.BatchNorm1d(50),
          nn.ReLU(),
          nn.MaxPool1d(8))
    self.fc = nn.Linear(300,6)
  def forward(self,x):
    #input.shape:(16,1,425)
    out = self.layer1(x)
    out = out.view(out.size(0),-1)
    out = self.fc(out)
    return out

输入的数据格式是(batch_size,word_vector,sequence_length),我设置的batch=16,特征工程样本是1x425,套用该格式就应该是(16,1,425)。对应nn.Conv1d的in_channels=1,out_channels就是你自己设置的,我选择的是100。

因为我做的是分类场景,所以做完两次一维卷积后还要加上一个线性层。

以上这篇pytorch中nn.Conv1d的用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python重试装饰器示例
Feb 11 Python
python压缩文件夹内所有文件为zip文件的方法
Jun 20 Python
Python中的pack和unpack的使用
Mar 12 Python
opencv改变imshow窗口大小,窗口位置的方法
Apr 02 Python
django 开发忘记密码通过邮箱找回功能示例
Apr 17 Python
浅谈Pandas:Series和DataFrame间的算术元素
Dec 22 Python
基于Numpy.convolve使用Python实现滑动平均滤波的思路详解
May 16 Python
Kali Linux安装ipython2 和 ipython3的方法
Jul 11 Python
python实现人像动漫化的示例代码
May 17 Python
Pycharm添加虚拟解释器报错问题解决方案
Oct 13 Python
详解python百行有效代码实现汉诺塔小游戏(简约版)
Oct 30 Python
OpenCV-Python模板匹配人眼的实例
Jun 08 Python
Python实现剪刀石头布小游戏(与电脑对战)
Dec 31 #Python
Pytorch之卷积层的使用详解
Dec 31 #Python
Python中bisect的使用方法
Dec 31 #Python
pytorch中tensor张量数据类型的转化方式
Dec 31 #Python
Pytorch之parameters的使用
Dec 31 #Python
使用TensorFlow-Slim进行图像分类的实现
Dec 31 #Python
Pytorch之view及view_as使用详解
Dec 31 #Python
You might like
php中对2个数组相加的函数
2011/06/24 PHP
解析php类的注册与自动加载
2013/07/05 PHP
让CodeIgniter的ellipsize()支持中文截断的方法
2014/06/12 PHP
thinkphp微信开之安全模式消息加密解密不成功的解决办法
2015/12/02 PHP
PHP实现的进度条效果详解
2016/05/03 PHP
IE6不能修改NAME问题的解决方法
2010/09/03 Javascript
toggle()隐藏问题的解决方法
2014/02/17 Javascript
js实现两点之间画线的方法
2015/05/12 Javascript
怎么通过onclick事件获取js函数返回值(代码少)
2015/07/28 Javascript
在WordPress中加入Google搜索功能的简单步骤讲解
2016/01/04 Javascript
Bootstrap的fileinput插件实现多文件上传的方法
2016/09/05 Javascript
JavaScript常用代码书写规范的超全面总结
2016/09/11 Javascript
基于twbsPagination.js分页插件使用心得(分享)
2017/10/21 Javascript
vue 组件开发原理与实现方法详解
2019/11/29 Javascript
Vue组件化开发之通用型弹出框的实现
2020/02/28 Javascript
Python字符串逐字符或逐词反转方法
2015/05/21 Python
Python自定义类的数组排序实现代码
2016/08/28 Python
详解 Python中LEGB和闭包及装饰器
2017/08/03 Python
机器学习之KNN算法原理及Python实现方法详解
2018/07/09 Python
python 输入一个数n,求n个数求乘或求和的实例
2018/11/13 Python
Python列表list常用内建函数实例小结
2019/10/22 Python
python 3.8.3 安装配置图文教程
2020/05/21 Python
Python远程方法调用实现过程解析
2020/07/28 Python
韩都衣舍天猫官方旗舰店:天猫女装销售总冠军
2017/10/10 全球购物
数百万免费的图形资源:Freepik
2020/09/21 全球购物
大学在校生求职信范文
2013/11/21 职场文书
中文师范生自荐信
2014/01/30 职场文书
建议书怎么写
2014/03/12 职场文书
2014年环境卫生工作总结
2014/11/24 职场文书
2014幼儿园教育教学工作总结
2014/12/17 职场文书
超市督导岗位职责
2015/04/10 职场文书
工作简报怎么写
2015/07/21 职场文书
《最后一头战象》教学反思
2016/02/16 职场文书
民事调解协议书
2016/03/21 职场文书
读《瓦尔登湖》有感:每个人都需要一个瓦尔登湖
2019/10/17 职场文书
vue中控制mock在开发环境使用,在生产环境禁用方式
2022/04/06 Vue.js