Pytorch mask_select 函數(shù)的用法詳解
非常簡單的函數(shù)，但是官網(wǎng)的介紹令人（令我）迷惑，所以稍加解釋。
mask_select會將滿足mask（掩碼、遮罩等等，隨便翻譯）的指示，將滿足條件的點選出來。
根據(jù)掩碼張量mask中的二元值，取輸入張量中的指定項( mask為一個 ByteTensor)，將取值返回到一個新的1D張量，
張量 mask須跟input張量有相同數(shù)量的元素數(shù)目，但形狀或維度不需要相同
x = torch.randn(3, 4)
x
1.2045 2.4084 0.4001 1.1372
0.5596 1.5677 0.6219 -0.7954
1.3635 -1.2313 -0.5414 -1.8478
[torch.FloatTensor of size 3x4]
mask = x.ge(0.5)
mask
1 1 0 1
1 1 1 0
1 0 0 0
[torch.ByteTensor of size 3x4]
torch.masked_select(x, mask)
1.2045
2.4084
1.1372
0.5596
1.5677
0.6219
1.3635
[torch.FloatTensor of size 7]
以上這篇Pytorch mask_select 函數(shù)的用法詳解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
- pytorch中的nn.ZeroPad2d()零填充函數(shù)實例詳解
- Pytorch 圖像變換函數(shù)集合小結
- Pytorch上下采樣函數(shù)--interpolate用法
- pytorch 常用函數(shù) max ,eq說明
- Pytorch十九種損失函數(shù)的使用詳解
- pytorch之Resize()函數(shù)具體使用詳解
- 使用 pytorch 創(chuàng)建神經(jīng)網(wǎng)絡擬合sin函數(shù)的實現(xiàn)
- PyTorch筆記之scatter()函數(shù)的使用
- pytorch方法測試——激活函數(shù)(ReLU)詳解
- pytorch 常用線性函數(shù)詳解
- 使用Pytorch來擬合函數(shù)方式
- 如何利用Pytorch計算三角函數(shù)
相關文章
這篇文章主要為大家介紹了python調(diào)用matlab，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-11-11
這篇文章主要介紹了python中的項目目錄結構,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
Python函數(shù)可變參數(shù)定義及其參數(shù)傳遞方式實例詳解
這篇文章主要介紹了Python函數(shù)可變參數(shù)定義及其參數(shù)傳遞方式,以實例形式較為詳細的分析了Python函數(shù)參數(shù)的使用技巧,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了Python獲取當前函數(shù)名稱方法實例分享，具有一定借鑒價值2018-01-01
這篇文章主要介紹了Python如何爬取qq音樂歌詞到本地,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
在本篇文章里小編給大家整理了關于python中聲明變量類型的相關知識點，需要的朋友們可以學習下。2020-06-06
今天小編就為大家分享一篇python自動識別文本編碼格式代碼，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
這篇文章主要介紹了關于python實現(xiàn)常用的相似度計算方法,最初的相似度計算是為了表征向量的重合程度的，在這里最經(jīng)典的就是余弦相似度了，當然使用正弦或者是正切等等三角函數(shù)也都是可以的,需要的朋友可以參考下2023-07-07
python光學仿真學習wxpython創(chuàng)建手速測試程序
這篇文章主要介紹了python光學仿真學習使用wxpython創(chuàng)建一個手速測試程序示例的實現(xiàn)，有需要的朋友可以借鑒參考下，希望能夠有所幫助2021-10-10
Python中多線程thread與threading的實現(xiàn)方法
這篇文章主要介紹了Python中多線程thread與threading的實現(xiàn)方法,很重要的應用,需要的朋友可以參考下2014-08-08