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发送email的3种方法
Apr 28 Python
Python设置Socket代理及实现远程摄像头控制的例子
Nov 13 Python
详解Python使用simplejson模块解析JSON的方法
Mar 24 Python
python解决Fedora解压zip时中文乱码的方法
Sep 18 Python
pandas apply 函数 实现多进程的示例讲解
Apr 20 Python
Python利用正则表达式实现计算器算法思路解析
Apr 25 Python
基于DataFrame筛选数据与loc的用法详解
May 18 Python
Python解决两个整数相除只得到整数部分的实例
Nov 10 Python
Django项目后台不挂断运行的方法
Aug 31 Python
利用python Selenium实现自动登陆京东签到领金币功能
Oct 31 Python
基于Python编写一个计算器程序,实现简单的加减乘除和取余二元运算
Aug 05 Python
python获取时间戳的实现示例(10位和13位)
Sep 23 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中用date函数获取当前时间有误的解决办法
2013/08/02 PHP
ThinkPHP中使用ajax接收json数据的方法
2014/12/18 PHP
Ext.FormPanel 提交和 Ext.Ajax.request 异步提交函数的区别
2009/11/12 Javascript
JQuery EasyUI 对话框的使用方法
2010/10/24 Javascript
js获得网页背景色和字体色的方法
2014/03/21 Javascript
jQuery表单域选择器用法分析
2015/02/10 Javascript
jquery使用on绑定a标签无效 只能用live解决
2016/06/02 Javascript
vue router自动判断左右翻页转场动画效果
2017/10/10 Javascript
Vue.js简易安装和快速入门(第二课)
2017/10/17 Javascript
jQuery Datatables表头不对齐的解决办法
2017/11/27 jQuery
Vue中的作用域CSS和CSS模块的区别
2018/10/09 Javascript
element vue Array数组和Map对象的添加与删除操作
2018/11/14 Javascript
JavaScript设计模式之命令模式实例分析
2019/01/16 Javascript
Quasar Input:type="number" 去掉上下小箭头 实现加减按钮样式功能
2020/04/09 Javascript
[06:33]3.19 DOTA2发布会 海涛、冷冷、2009见证希望
2014/03/21 DOTA
[02:30]辉夜杯主赛事第二日胜者组半决赛 CDEC.Y赛后采访
2015/12/26 DOTA
python字符串加密解密的三种方法分享(base64 win32com)
2014/01/19 Python
在Python中使用Mako模版库的简单教程
2015/04/08 Python
Python学习小技巧之列表项的拼接
2017/05/20 Python
python sort、sort_index方法代码实例
2019/03/28 Python
Python2.7版os.path.isdir中文路径返回false的解决方法
2019/06/21 Python
用python求一重积分和二重积分的例子
2019/12/06 Python
Keras—embedding嵌入层的用法详解
2020/06/10 Python
如何清空python的变量
2020/07/05 Python
Python 忽略文件名编码的方法
2020/08/01 Python
python 基于opencv 绘制图像轮廓
2020/12/11 Python
StubHub德国:购买和出售门票
2017/09/06 全球购物
size?德国官方网站:英国伦敦的球鞋精品店
2018/03/17 全球购物
某IT外企面试题-二分法求方程!看看大家的C++功底
2015/07/04 面试题
纪委书记群众路线整改措施思想汇报
2014/10/09 职场文书
销售员态度差检讨书
2014/10/26 职场文书
2014个人年终工作总结范文
2014/12/15 职场文书
2014年学校团委工作总结
2014/12/20 职场文书
宣传稿格式范文
2015/07/23 职场文书
学校中层领导培训心得体会
2016/01/11 职场文书
交互式可视化js库gojs使用介绍及技巧
2022/02/18 Javascript