pytorch forward兩個參數(shù)實(shí)例
更新時間：2020年01月17日 09:32:18 作者：weixin_41950276
今天小編就為大家分享一篇pytorch forward兩個參數(shù)實(shí)例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
以channel Attention Block為例子
class CAB(nn.Module):
def __init__(self, in_channels, out_channels):
super(CAB, self).__init__()
self.global_pooling = nn.AdaptiveAvgPool2d(output_size=1)
self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=1, stride=1, padding=0)
self.relu = nn.ReLU()
self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=1, stride=1, padding=0)
self.sigmod = nn.Sigmoid()
def forward(self, x):
x1, x2 = x # high, low
x = torch.cat([x1,x2],dim=1)
x = self.global_pooling(x)
x = self.conv1(x)
x = self.relu(x)
x = self.conv2(x)
x = self.sigmod(x)
x2 = x * x2
res = x2 + x1
return res
以上這篇pytorch forward兩個參數(shù)實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
如何利用Python解析超大的json數(shù)據(jù)(GB級別)
果不想從頭開始創(chuàng)建數(shù)據(jù)格式來存儲數(shù)據(jù),JSON是一個不錯的選擇,下面這篇文章主要給大家介紹了關(guān)于如何利用Python解析超大的json數(shù)據(jù)(GB級別)的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-07-07
python執(zhí)行scp命令拷貝文件及文件夾到遠(yuǎn)程主機(jī)的目錄方法
今天小編就為大家分享一篇python執(zhí)行scp命令拷貝文件及文件夾到遠(yuǎn)程主機(jī)的目錄方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
這篇文章主要介紹了python如何做代碼性能分析，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下2021-04-04
使用Python中的自帶庫math、自帶函數(shù)pow和自帶庫cmath來對數(shù)字進(jìn)行開根號運(yùn)算，這篇文章主要介紹了Python開根號的幾種方式,需要的朋友可以參考下2023-01-01
在本篇文章里小編給大家整理的是一篇關(guān)于python中re模塊知識點(diǎn)總結(jié)內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。2021-01-01
這篇文章主要介紹了pytorch 中nn.Dropout的使用說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
CentOS上安裝的python版本是2.6，不能滿足我運(yùn)行軟件的要求，所以對python進(jìn)行升級。Python的最新版本已經(jīng)是3.3，但是Python3的兼容性可能還有一定的問題，所以還是升級到2.7較為保險(xiǎn)。2014-07-07
這篇文章主要為大家介紹了Python多進(jìn)程使用及進(jìn)程池詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-06-06
python使用xslt提取網(wǎng)頁數(shù)據(jù)的方法
這篇文章主要為大家詳細(xì)介紹了Python使用xslt提取網(wǎng)頁數(shù)據(jù)的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-02-02
這篇文章主要介紹了Python?matplotlib底層原理，下面文章圍繞Python?matplotlib底層原理的相關(guān)資料展開詳細(xì)內(nèi)容,具有一定的參考價值，需要的朋友可以參考下2021-12-12