關(guān)于nn.BatchNorm1d()用法及說(shuō)明
nn.BatchNorm1d()用法
torch.nn.BatchNorm1d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
num_features– 特征維度eps– 為數(shù)值穩(wěn)定性而加到分母上的值。momentum– 移動(dòng)平均的動(dòng)量值。affine–一個(gè)布爾值，當(dāng)設(shè)置為真時(shí)，此模塊具有可學(xué)習(xí)的仿射參數(shù)。
這里其他幾個(gè)參數(shù)都不重要，只需要看num_features就可以了。
num_features就是你需要?dú)w一化的那一維的維度。
nn.BatchNorm1d本身不是給定輸入矩陣，輸出歸一化結(jié)果的函數(shù)，而是定義了一個(gè)方法，再用這個(gè)方法去做歸一化。
BN = nn.BatchNorm1d(100) input = torch.randn(20, 100) output = m(input)
我們首先定義了一個(gè)歸一化的函數(shù)BN，需要?dú)w一化的維度為100，其他參數(shù)為默認(rèn)。
然后隨機(jī)初始化一個(gè)20×100的矩陣input，再用BN對(duì)這個(gè)矩陣歸一化。
函數(shù)的input可以是二維或者三維。
- 當(dāng)input的維度為（N, C）時(shí)，BN將對(duì)C維歸一化；
- 當(dāng)input的維度為(N, C, L) 時(shí)，歸一化的維度同樣為C維。
使用nn.BatchNorm1d出現(xiàn)RuntimeError
pytorch模型中添加batchnorm過(guò)程中出現(xiàn)錯(cuò)誤
RuntimeError: running_mean should contain 1 elements not 512
錯(cuò)誤來(lái)自于nn.BatchNorm1d(N)，其中N=channels
解決辦法
將輸入重新排列
video = video.view(-1, video.shape[-1])
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了django中的*args 與 **kwargs使用介紹，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-03-03
Python字符串的字符轉(zhuǎn)換、字符串劈分、字符串合并問(wèn)題分析
這篇文章主要介紹了Python字符串的字符轉(zhuǎn)換、字符串劈分、字符串合并,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2023-03-03
Tensorflow 實(shí)現(xiàn)分批量讀取數(shù)據(jù)
今天小編就為大家分享一篇Tensorflow 實(shí)現(xiàn)分批量讀取數(shù)據(jù)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-01-01
PyQt5.6+pycharm配置以及pyinstaller生成exe(小白教程)
這篇文章主要介紹了PyQt5.6+pycharm配置以及pyinstaller生成exe，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06
這篇文章主要介紹了Django顯示可視化圖表的實(shí)踐，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-05-05
python 使用pandas同時(shí)對(duì)多列進(jìn)行賦值
這篇文章主要介紹了python 使用pandas同時(shí)對(duì)多列進(jìn)行賦值操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-03-03
python中whl文件封裝的實(shí)現(xiàn)示例
Wheel（.whl）是 Python 的一種內(nèi)置包格式,本文就來(lái)介紹一下python中whl文件封裝的實(shí)現(xiàn)示例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2025-08-08
Python實(shí)現(xiàn)提取圖片中顏色并繪制成可視化圖表
今天小編來(lái)為大家分享一個(gè)有趣的可視化技巧，就是如何利用Python語(yǔ)言實(shí)現(xiàn)從圖片中提取顏色然后繪制成可視化圖表，感興趣的可以嘗試一下2022-07-07
簡(jiǎn)單瞅瞅Python vars()內(nèi)置函數(shù)的實(shí)現(xiàn)
這篇文章主要介紹了簡(jiǎn)單瞅瞅Python vars()內(nèi)置函數(shù)的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09
這篇文章詳細(xì)介紹了?Python?常見的編碼和解碼技術(shù),包括字符編碼（如?ASCII、UTF-8?等）、數(shù)據(jù)編碼（如?Base64、URL?編碼等）、文件編碼,通過(guò)代碼示例講解操作方法及錯(cuò)誤處理,還給出了常見編碼的選擇指南,強(qiáng)調(diào)掌握這些技術(shù)對(duì)編寫程序很重要,需要的朋友可以參考下2025-01-01