溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

如何在Pytorch中使用mask_select 函數(shù)

發(fā)布時(shí)間：2021-01-11 15:50:51 來(lái)源：億速云 閱讀：560 作者：Leah 欄目：開發(fā)技術(shù)

今天就跟大家聊聊有關(guān)如何在Pytorch中使用mask_select 函數(shù)，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結(jié)了以下內(nèi)容，希望大家根據(jù)這篇文章可以有所收獲。

非常簡(jiǎn)單的函數(shù)，但是官網(wǎng)的介紹令人（令我）迷惑，所以稍加解釋。

mask_select會(huì)將滿足mask（掩碼、遮罩等等，隨便翻譯）的指示，將滿足條件的點(diǎn)選出來(lái)。

根據(jù)掩碼張量mask中的二元值，取輸入張量中的指定項(xiàng)( mask為一個(gè) ByteTensor)，將取值返回到一個(gè)新的1D張量，

張量 mask須跟input張量有相同數(shù)量的元素?cái)?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]

看完上述內(nèi)容，你們對(duì)如何在Pytorch中使用mask_select 函數(shù)有進(jìn)一步的了解嗎？如果還想了解更多知識(shí)或者相關(guān)內(nèi)容，請(qǐng)關(guān)注億速云行業(yè)資訊頻道，感謝大家的支持。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手