淺談Pytorch中的torch.gather函數(shù)的含義
更新時間：2019年08月18日 17:09:17 作者：江戶川柯壯
今天小編就為大家分享一篇淺談Pytorch中的torch.gather函數(shù)的含義，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch中的gather函數(shù)
pytorch比tensorflow更加編程友好，所以準備用pytorch試著做最近要做的一些實驗。
立個flag開始學(xué)習(xí)pytorch，新開一個分類整理學(xué)習(xí)pytorch中的一些踩到的泥坑。
今天剛開始接觸，讀了一下documentation，寫一個一開始每太搞懂的函數(shù)gather
b = torch.Tensor([[1,2,3],[4,5,6]]) print b index_1 = torch.LongTensor([[0,1],[2,0]]) index_2 = torch.LongTensor([[0,1,1],[0,0,0]]) print torch.gather(b, dim=1, index=index_1) print torch.gather(b, dim=0, index=index_2)
觀察它的輸出結(jié)果：
1 2 3 4 5 6 [torch.FloatTensor of size 2x3] 1 2 6 4 [torch.FloatTensor of size 2x2] 1 5 6 1 2 3 [torch.FloatTensor of size 2x3]
這里是官方文檔的解釋
torch.gather(input, dim, index, out=None) → Tensor Gathers values along an axis specified by dim. For a 3-D tensor the output is specified by: out[i][j][k] = input[index[i][j][k]][j][k] # dim=0 out[i][j][k] = input[i][index[i][j][k]][k] # dim=1 out[i][j][k] = input[i][j][index[i][j][k]] # dim=2 Parameters: input (Tensor) – The source tensor dim (int) – The axis along which to index index (LongTensor) – The indices of elements to gather out (Tensor, optional) – Destination tensor Example: >>> t = torch.Tensor([[1,2],[3,4]]) >>> torch.gather(t, 1, torch.LongTensor([[0,0],[1,0]])) 1 1 4 3 [torch.FloatTensor of size 2x2]
可以看出，gather的作用是這樣的，index實際上是索引，具體是行還是列的索引要看前面dim 的指定，比如對于我們的栗子，【1,2,3;4,5,6,】，指定dim=1，也就是橫向，那么索引就是列號。index的大小就是輸出的大小，所以比如index是【1,0;0,0】，那么看index第一行，1列指的是2， 0列指的是1，同理，第二行為4，4 。這樣就輸入為【2,1;4,4】，參考這樣的解釋看上面的輸出結(jié)果，即可理解gather的含義。
gather在one-hot為輸出的多分類問題中，可以把最大值坐標作為index傳進去，然后提取到每一行的正確預(yù)測結(jié)果，這也是gather可能的一個作用。
以上這篇淺談Pytorch中的torch.gather函數(shù)的含義就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
基于Python的自媒體小助手---登錄頁面的實現(xiàn)代碼
這篇文章主要介紹了基于Python的自媒體小助手---登錄頁面的實現(xiàn)代碼，本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-06-06
今天小編就為大家分享一篇python自定義線程池控制線程數(shù)量的示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02
求和是數(shù)學(xué)中最基本的運算之一,也是編程中常見的任務(wù)之一,Python 提供了多種方法來計算和求和數(shù)字,本文將掏出計算求和的不同方法,包括使用循環(huán)、內(nèi)置函數(shù)以及第三方庫2023-11-11
本文給大家分享Python 語言實現(xiàn)六大查找算法，針對每種算法通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友參考下吧2021-06-06
Python工程師面試題 與Python基礎(chǔ)語法相關(guān)
這篇文章主要為大家分享了Python工程師面試題，面試題的內(nèi)容主要與Python基礎(chǔ)語法相關(guān)，感興趣的小伙伴們可以參考一下2016-01-01
這篇文章主要介紹了python使用PIL給圖片添加文字生成海報示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-08-08
13行python代碼實現(xiàn)對微信進行推送消息的示例代碼
本文主要介紹了13行python代碼實現(xiàn)對微信進行推送消息的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
本文主要介紹了NumPy內(nèi)存布局的實現(xiàn),括連續(xù)內(nèi)存布局（C順序）和分散內(nèi)存布局（Fortran順序）,并通過實例演示如何操作數(shù)組的內(nèi)存布局,感興趣的可以了解一下2024-01-01
這篇文章主要介紹了在Python中使用next()方法操作文件的教程,是Python入門中的基礎(chǔ)知識,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了python中的編碼和解碼及\x和\u問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05