pytorch中nn.Conv1d的用法詳解
更新時(shí)間：2019年12月31日 11:13:57 作者：好運(yùn)帶來喜和愛
今天小編就為大家分享一篇pytorch中nn.Conv1d的用法詳解，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
先粘貼一段official guide：nn.conv1d官方
我一開始被in_channels、out_channels卡住了很久，結(jié)果發(fā)現(xiàn)就和conv2d是一毛一樣的。話不多說，先粘代碼（菜雞的自我修養(yǎng)）
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
輸入的數(shù)據(jù)格式是（batch_size,word_vector,sequence_length），我設(shè)置的batch=16，特征工程樣本是1x425，套用該格式就應(yīng)該是（16，1，425）。對應(yīng)nn.Conv1d的in_channels=1，out_channels就是你自己設(shè)置的，我選擇的是100。
因?yàn)槲易龅氖欠诸悎鼍?，所以做完兩次一維卷積后還要加上一個(gè)線性層。
以上這篇pytorch中nn.Conv1d的用法詳解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
最近有部分小伙伴問我,python 寫的項(xiàng)目可不可以打包成exe程序,放到?jīng)]有python環(huán)境上的電腦中執(zhí)行? 答案當(dāng)然是可以的,下面這篇文章主要給大家介紹了如何通過一篇文章學(xué)會兩種將pyton打包成exe的方式,需要的朋友可以參考下2021-11-11
windows上安裝python3教程以及環(huán)境變量配置詳解
這篇文章主要介紹了windows上安裝python3教程以及環(huán)境變量配置詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Python中的賦值運(yùn)算符,其使用是Python的基本功,需要的朋友可以參考下2015-05-05
python+matplotlib實(shí)現(xiàn)鼠標(biāo)移動三角形高亮及索引顯示
這篇文章主要介紹了Python+matplotlib實(shí)現(xiàn)鼠標(biāo)移動三角形高亮及索引顯示，具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
今天小編就為大家分享一篇python 自動軌跡繪制的實(shí)例代碼，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
大家好，本篇文章主要講的是python的圖形用戶界面介紹，感興趣的同學(xué)趕快來看一看吧，對你有幫助的話記得收藏一下，方便下次瀏覽2022-01-01
Python pydotplus安裝及可視化圖形創(chuàng)建教程
這篇文章主要為大家介紹了Python pydotplus安裝及可視化圖形創(chuàng)建教程示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-10-10
這篇文章主要為大家詳細(xì)介紹了python制作圖片縮略圖的相關(guān)方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-04-04
venv 是 Python 內(nèi)置標(biāo)準(zhǔn)庫中創(chuàng)建輕量級虛擬環(huán)境的工具,本文通過示例代碼介紹python venv和virtualenv的相關(guān)知識,感興趣的朋友跟隨小編一起看看吧2024-08-08
python實(shí)現(xiàn)可下載音樂的音樂播放器
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)可下載音樂的音樂播放器，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-02-02