pytorch中nn.Conv1d的用法详解
更新时间：2019年12月31日 11:13:57 作者：好运带来喜和爱
今天小编就为大家分享一篇pytorch中nn.Conv1d的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
先粘贴一段official guide：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 写的项目可不可以打包成exe程序,放到没有python环境上的电脑中执行? 答案当然是可以的,下面这篇文章主要给大家介绍了如何通过一篇文章学会两种将pyton打包成exe的方式,需要的朋友可以参考下2021-11-11
这篇文章主要介绍了windows上安装python3教程以及环境变量配置详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了Python中的赋值运算符,其使用是Python的基本功,需要的朋友可以参考下2015-05-05
python+matplotlib实现鼠标移动三角形高亮及索引显示
这篇文章主要介绍了Python+matplotlib实现鼠标移动三角形高亮及索引显示，具有一定借鉴价值,需要的朋友可以参考下2018-01-01
今天小编就为大家分享一篇python 自动轨迹绘制的实例代码，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
大家好，本篇文章主要讲的是python的图形用户界面介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2022-01-01
这篇文章主要为大家介绍了Python pydotplus安装及可视化图形创建教程示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-10-10
这篇文章主要为大家详细介绍了python制作图片缩略图的相关方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-04-04
venv 是 Python 内置标准库中创建轻量级虚拟环境的工具,本文通过示例代码介绍python venv和virtualenv的相关知识,感兴趣的朋友跟随小编一起看看吧2024-08-08
这篇文章主要为大家详细介绍了python实现可下载音乐的音乐播放器，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-02-02
最新评论