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实现模拟分割大文件及多线程处理的方法,涉及Python文件读取、分割及多线程相关操作技巧,需要的朋友可以参考下2017-10-10
Apscheduler是基于Quartz的Python定时任务框架，功能上跟Quartz一致，使用上跟Quartz也几乎一致。下面通过本文给大家介绍Python定时库Apscheduler的简单使用，感兴趣的朋友一起看看吧2021-11-11
这篇文章主要介绍了Python Selenium如何切换浏览器的页面的相关资料,在使用Selenium进行网页测试时,跳转页面后常会出现无法定位元素的问题,解决这一问题的关键是学会在多个浏览器标签页或窗口间切换,需要的朋友可以参考下2024-10-10
Python中类方法@classmethod和静态方法@staticmethod解析
这篇文章主要介绍了Python中类方法@classmethod和静态方法@staticmethod解析,python中存在三种方法,分别为常规方法（定义中传入self）、@classmethod修饰的类方法、@staticmethod修饰的静态方法,,需要的朋友可以参考下2023-08-08
这篇文章主要为大家详细介绍了python实现坦克大战，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-04-04
这篇文章主要介绍了Python基于read(size)方法读取超大文件,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
这篇文章主要给大家介绍了关于python中list.copy方法使用的相关资料,文中还介绍了python list.copy() 和 copy.deepcopy()区别,需要的朋友可以参考下2023-02-02
Python自定义聚合函数merge与transform区别详解
这篇文章主要介绍了Python自定义聚合函数merge与transform区别详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-05-05
本文主要介绍了Python中的type与isinstance的区别详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
本篇文章主要介绍了python3实现的web端json通信协议，具有一定的参考价值，感兴趣的小伙伴们可以参考一下。2016-12-12
最新评论