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中的闭包总结
Sep 18 Python
Python用threading实现多线程详解
Feb 03 Python
基于Python3 逗号代码 和 字符图网格(详谈)
Jun 22 Python
python3操作mysql数据库的方法
Jun 23 Python
Python xlwt设置excel单元格字体及格式
Apr 18 Python
Python操作word常见方法示例【win32com与docx模块】
Jul 17 Python
Python3多目标赋值及共享引用注意事项
May 27 Python
关于python字符串方法分类详解
Aug 20 Python
python破解bilibili滑动验证码登录功能
Sep 11 Python
浅谈python出错时traceback的解读
Jul 15 Python
python基础之错误和异常处理
Oct 24 Python
 分享一个Python 遇到数据库超好用的模块
Apr 06 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
Codeigniter实现智能裁剪图片的方法
2014/06/12 PHP
PHP类的自动加载与命名空间用法实例分析
2020/06/05 PHP
Javascript类库的顶层对象名用户体验分析
2010/10/24 Javascript
javaScript 页面自动加载事件详解
2014/02/10 Javascript
jQuery设置与获取HTML,文本和值的简单实例
2014/02/26 Javascript
nodejs npm package.json中文文档
2014/09/04 NodeJs
使用JQuery实现智能表单验证功能
2016/03/08 Javascript
javascript原型继承工作原理和实例详解
2016/04/07 Javascript
HTML5 JS压缩图片并获取图片BASE64编码上传
2020/11/16 Javascript
基于javascript的Form表单验证
2016/12/29 Javascript
详解extract-text-webpack-plugin 的使用及安装
2018/06/12 Javascript
vue3.0 CLI - 2.1 -  component 组件入门教程
2018/09/14 Javascript
微信小程序MUI导航栏透明渐变功能示例(通过改变opacity实现)
2019/01/24 Javascript
jQuery HTML设置内容和属性操作实例分析
2020/05/20 jQuery
浅谈使用nodejs搭建web服务器的过程
2020/07/20 NodeJs
[01:13]这,就是刀塔
2014/07/16 DOTA
比较详细Python正则表达式操作指南(re使用)
2008/09/06 Python
python使用rabbitmq实现网络爬虫示例
2014/02/20 Python
python实现中文分词FMM算法实例
2015/07/10 Python
Python实现将Excel转换为json的方法示例
2017/08/05 Python
Python学习之Anaconda的使用与配置方法
2018/01/04 Python
python 移除字符串尾部的数字方法
2018/07/17 Python
对python3 中方法各种参数和返回值详解
2018/12/15 Python
Python3如何对urllib和urllib2进行重构
2019/11/25 Python
Python __slots__的使用方法
2020/11/15 Python
职业技术学校毕业生推荐信
2013/12/03 职场文书
活动总结怎么写
2014/04/28 职场文书
室内设计专业自荐信
2014/05/31 职场文书
采购部长岗位职责
2014/06/13 职场文书
2014年向国旗敬礼活动总结
2014/09/27 职场文书
小学优秀学生评语
2014/12/29 职场文书
导游欢迎词范文
2015/01/23 职场文书
2015年端午节活动策划书
2015/05/05 职场文书
公司周年庆寄语
2019/06/21 职场文书
分享Python获取本机IP地址的几种方法
2022/03/17 Python
python中pd.cut()与pd.qcut()的对比及示例
2022/06/16 Python