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是编译运行的验证方法
Jan 30 Python
在Python的Django框架下使用django-tagging的教程
May 30 Python
python学习笔记之调用eval函数出现invalid syntax错误问题
Oct 18 Python
对python内置map和six.moves.map的区别详解
Dec 19 Python
Python 动态导入对象,importlib.import_module()的使用方法
Aug 28 Python
python制作朋友圈九宫格图片
Nov 03 Python
PyTorch中反卷积的用法详解
Dec 30 Python
python对数组进行排序,并输出排序后对应的索引值方式
Feb 28 Python
vue常用指令代码实例总结
Mar 16 Python
解析python 中/ 和 % 和 //(地板除)
Jun 28 Python
Python eval函数原理及用法解析
Nov 14 Python
jupyter notebook更换皮肤主题的实现
Jan 07 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
地球防卫队:陪着奥特曼打小怪兽的人类力量 那些经典队服
2020/03/08 日漫
轻松入门: 煮好咖啡的七个诀窍
2021/03/03 冲泡冲煮
php中常用字符串处理代码片段整理
2011/11/07 PHP
PHP 线程安全与非线程安全版本的区别深入解析
2013/08/06 PHP
PHP 错误处理机制
2015/07/06 PHP
详解WordPress中提醒安装插件以及隐藏插件的功能实现
2015/12/25 PHP
Yii2中使用join、joinwith多表关联查询
2016/06/30 PHP
PHP实现生成带背景的图形验证码功能
2016/10/03 PHP
阿里云Win2016安装Apache和PHP环境图文教程
2018/03/11 PHP
php和asp语法上的区别总结
2019/05/12 PHP
js类中获取外部函数名的方法
2007/08/19 Javascript
js 未结束的字符串常量错误解决方法
2010/06/13 Javascript
计算世界完全对称日的js代码,粗糙版
2011/11/04 Javascript
jquery实现弹出div,始终显示在屏幕正中间的简单实例
2014/03/08 Javascript
JavaScript设计模式之代理模式介绍
2014/12/28 Javascript
JavaScript简单表格编辑功能实现方法
2015/04/16 Javascript
JAVASCRIPT代码编写俄罗斯方块网页版
2015/11/26 Javascript
JS获取复选框的值,并传递到后台的实现方法
2016/05/30 Javascript
DOM操作和jQuery实现选项移动操作的简单实例
2016/06/07 Javascript
Form表单按回车自动提交表单的实现方法
2016/11/18 Javascript
Easy UI动态树点击文字实现展开关闭功能
2017/09/30 Javascript
node中的cookie的具体使用
2018/09/13 Javascript
es6 symbol的实现方法示例
2019/04/02 Javascript
详解django模板与vue.js冲突问题
2019/07/07 Javascript
node.js中对Event Loop事件循环的理解与应用实例分析
2020/02/14 Javascript
解决python flask中config配置管理的问题
2019/07/26 Python
Python for循环搭配else常见问题解决
2020/02/11 Python
CSS3绘制超炫的上下起伏波动进度加载动画
2016/04/21 HTML / CSS
美国从事品牌鞋类零售的连锁店:Famous Footwear
2016/08/25 全球购物
德国著名廉价网上药店:Shop-Apotheke
2017/07/23 全球购物
美国价格实惠的在线眼镜网站:Zeelool
2020/12/25 全球购物
当文件系统受到破坏时,如何检查和修复系统?
2012/03/09 面试题
建筑横幅标语
2014/10/09 职场文书
单位委托书
2014/10/15 职场文书
2015年妇女工作总结
2015/05/14 职场文书
golang中的并发和并行
2021/05/08 Golang