解析Pytorch中的torch.gather()函數(shù)
更新時(shí)間：2021年11月13日 13:58:40 作者：xiaoliujun1999
本文給大家介紹了Pytorch中的torch.gather()函數(shù)，通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友參考下吧
參數(shù)說(shuō)明
以官方說(shuō)明為例，gather()函數(shù)需要三個(gè)參數(shù)，輸入input，維度dim，以及索引index
input必須為T(mén)ensor類(lèi)型
dim為int類(lèi)型，代表從哪個(gè)維度進(jìn)行索引
index為L(zhǎng)ongTensor類(lèi)型
舉例說(shuō)明
input=torch.tensor([[1,2,3],[4,5,6]]) #作為輸入 index1=torch.tensor([[0,1,1],[0,1,1]]) #作為索引矩陣 # dim=0時(shí),按列進(jìn)行索引 print (torch.gather(input,dim=0,index=index1)) # dim=1時(shí),按行進(jìn)行索引 print (torch.gather(input,dim=1,index=index1))
結(jié)果如下圖所示：
# 按列進(jìn)行索引
tensor([[1, 5, 6],
[4, 2, 6]])
# 按行進(jìn)行索引
tensor([[1, 2, 2],
[5, 4, 5]])
畫(huà)圖說(shuō)明
官方文檔
def gather(self, input, dim, index, *args, **kwargs):
For a 3-D tensor the output is specified by::
out[i][j][k] = input[index[i][j][k]][j][k] # if dim == 0
out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1
out[i][j][k] = input[i][j][index[i][j][k]] # if dim == 2
Args:
input (Tensor): the source tensor
dim (int): the axis along which to index
index (LongTensor): the indices of elements to gather
Example::
>>> t = torch.tensor([[1, 2], [3, 4]])
>>> torch.gather(t, 1, torch.tensor([[0, 0], [1, 0]]))
tensor([[ 1, 1],
[ 4, 3]])
到此這篇關(guān)于Pytorch中的torch.gather()函數(shù)的文章就介紹到這了,更多相關(guān)Pytorch torch.gather()函數(shù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
今天小編就為大家分享一篇python tkinter canvas 顯示圖片的示例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06
PyQt6+pyqtgraph折線(xiàn)圖繪制顯示功能
文章介紹了如何在PyQt6中使用pyqtgraph庫(kù)繪制和顯示折線(xiàn)圖,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),感興趣的朋友跟隨小編一起看看吧2024-11-11
這篇文章主要為大家詳細(xì)介紹了Flask是如何使用日志功能的，文中的示例代碼講解詳細(xì)，對(duì)我們深入了解Flask有一定的幫助，需要的可以參考一下2023-05-05
這篇文章主要為大家詳細(xì)介紹了tensorflow實(shí)現(xiàn)邏輯回歸模型的相關(guān)資料，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-09-09
Python標(biāo)準(zhǔn)庫(kù)之循環(huán)器(itertools)介紹
這篇文章主要介紹了Python標(biāo)準(zhǔn)庫(kù)之循環(huán)器(itertools)介紹,本文講解了無(wú)窮循環(huán)器、函數(shù)式工具、組合工具、groupby()、其它工具等內(nèi)容,需要的朋友可以參考下2014-11-11
python基于隱馬爾可夫模型實(shí)現(xiàn)中文拼音輸入
這篇文章主要介紹了python基于隱馬爾可夫模型實(shí)現(xiàn)中文拼音輸入的相關(guān)資料,需要的朋友可以參考下2016-04-04
詳解Django解決ajax跨域訪(fǎng)問(wèn)問(wèn)題
這篇文章主要介紹了詳解Django解決ajax跨域訪(fǎng)問(wèn)問(wèn)題，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-08-08
使用Scrapy框架爬取網(wǎng)頁(yè)并保存到Mysql的實(shí)現(xiàn)
本文主要介紹了使用Scrapy框架爬取網(wǎng)頁(yè)并保存到Mysql的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
這篇文章主要介紹了如何使用Python?VTK高亮顯示actor，通過(guò)Python-VTK在同一個(gè)窗口中，高亮顯示選中的actor。本例子中的代碼，當(dāng)窗口中的圓球actor被選中時(shí)，會(huì)變成紅色，并且會(huì)顯示actor三遍面片邊緣信息，下文相關(guān)內(nèi)容需要的小伙伴可以參考一下2022-04-04
python猜數(shù)字小游戲?qū)崿F(xiàn)代碼
大家好，本篇文章主要講的是python猜數(shù)字小游戲?qū)崿F(xiàn)代碼，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話(huà)記得收藏一下，方便下次瀏覽2022-01-01