pytorch forward兩個參數(shù)實例

更新時間：2020年01月17日 09:32:18 作者：weixin_41950276

今天小編就為大家分享一篇pytorch forward兩個參數(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ù)實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python實現(xiàn)把二維列表變?yōu)橐痪S列表的方法分析
這篇文章主要介紹了python實現(xiàn)把二維列表變?yōu)橐痪S列表的方法,結(jié)合實例形式總結(jié)分析了Python列表推導式、嵌套、循環(huán)等相關(guān)操作技巧,需要的朋友可以參考下
2019-10-10
一步步教你用Python畫五彩氣球
這篇文章主要給大家介紹了關(guān)于如何用Python畫五彩氣球的相關(guān)資料,主要是用turtle庫自帶的畫筆turtle.Turtle()來繪制氣球,文中給出了詳細的實例代碼,需要的朋友可以參考下
2023-06-06
Python實現(xiàn)一個簡單三層神經(jīng)網(wǎng)絡(luò)的搭建及測試代碼解析
一個完整的神經(jīng)網(wǎng)絡(luò)一般由三層構(gòu)成：輸入層，隱藏層（可以有多層）和輸出層。本文所構(gòu)建的神經(jīng)網(wǎng)絡(luò)隱藏層只有一層。一個神經(jīng)網(wǎng)絡(luò)主要由三部分構(gòu)成（代碼結(jié)構(gòu)上）：初始化，訓練，和預(yù)測。,需要的朋友可以參考下面文章內(nèi)容的具體內(nèi)容
2021-09-09
Python編程中被忽視的核心技巧總結(jié)
這篇文章主要介紹了一些在?Python?編程中可能被忽視的核心功能，包括默認參數(shù)、海象運算符、*args?和?**kwargs?的使用等，需要的可參考下
2023-08-08
python uuid生成唯一id或str的最簡單案例
這篇文章主要介紹了python uuid生成唯一id或str的最簡單案例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2021-01-01
Python安裝Numpy出現(xiàn)異常信息簡單解決辦法
在安裝Python的Numpy包時,可能會遇到路徑警告或包源超時的問題,首先,如果出現(xiàn)包源超時,可以嘗試更換為國內(nèi)的鏡像源,如清華大學鏡像源,其次,如果在安裝完成后提示將某個路徑添加到PATH環(huán)境變量,按照提示操作即可消除異常,需要的朋友可以參考下
2024-09-09
python中l(wèi)eastsq函數(shù)的使用方法
這篇文章主要介紹了python中l(wèi)eastsq函數(shù)的使用方法，leastsq作用是最小化一組方程的平方和，下面文章舉例說明詳細內(nèi)容，具有一的參考價值，需要的小伙伴可以參考一下
2022-03-03
WxPython開發(fā)之實現(xiàn)表格數(shù)據(jù)導出到Excel并打開
在 Python 中使用 wxPython 導出實體類列表數(shù)據(jù)到 Excel,通常可以借助 openpyxl 或 pandas 庫來實現(xiàn),下面就跟隨小編一起來了解下具體操作吧
2024-12-12
Python圖算法實例分析
這篇文章主要介紹了Python圖算法,結(jié)合實例形式詳細分析了Python數(shù)據(jù)結(jié)構(gòu)與算法中的圖算法實現(xiàn)技巧,需要的朋友可以參考下
2016-08-08
利用pandas將非數(shù)值數(shù)據(jù)轉(zhuǎn)換成數(shù)值的方式
今天小編就為大家分享一篇利用pandas將非數(shù)值數(shù)據(jù)轉(zhuǎn)換成數(shù)值的方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-12-12