淺談Pytorch中的torch.gather函數的含義
更新時間：2019年08月18日 17:09:17 作者：江戶川柯壯
今天小編就為大家分享一篇淺談Pytorch中的torch.gather函數的含義，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch中的gather函數
pytorch比tensorflow更加編程友好，所以準備用pytorch試著做最近要做的一些實驗。
立個flag開始學習pytorch，新開一個分類整理學習pytorch中的一些踩到的泥坑。
今天剛開始接觸，讀了一下documentation，寫一個一開始每太搞懂的函數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)
觀察它的輸出結果：
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】，參考這樣的解釋看上面的輸出結果，即可理解gather的含義。
gather在one-hot為輸出的多分類問題中，可以把最大值坐標作為index傳進去，然后提取到每一行的正確預測結果，這也是gather可能的一個作用。
以上這篇淺談Pytorch中的torch.gather函數的含義就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
在這個充滿浪漫的季節(jié),如何用代碼表達你的愛意呢？今天我們將使用 Python 的 matplotlib 和 numpy 庫繪制一個動態(tài)的愛心,并且在愛心上添加表白的文字,這將是一個獨特而浪漫的方式來表達你的心聲,感興趣的小伙伴跟著小編來看看吧2025-04-04
本篇文章將分享Prometheus?API的基本概念到PromQL查詢語言的應用,再到如何通過Python與Prometheus?API進行無縫交互,通過豐富的示例代碼和詳細的講解,將解鎖使用Python進行實時監(jiān)控的奇妙世界,為讀者打開更廣闊的數據分析視野2024-01-01
使用beaker讓Facebook的Bottle框架支持session功能
這篇文章主要介紹了使用beaker讓Facebook的Bottle框架支持session功能,session在Python的Django等框架中內置但在Bottle中并沒有被集成,需要的朋友可以參考下2015-04-04
Python模擬登陸淘寶并統(tǒng)計淘寶消費情況的代碼實例分享
借助urllib、urllib2和BeautifulSoup等幾個模塊的常用爬蟲開發(fā)組合,我們能夠輕易實現一份淘寶對賬單,這里我們就來看一則Python模擬登陸淘寶并統(tǒng)計淘寶消費情況的代碼實例分享:2016-07-07
這篇文章主要介紹了Python sqlite3查詢操作過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-02-02
這篇文章主要介紹了python函數的定義和調用,實例分析了Python中返回一個返回值與多個返回值的方法,需要的朋友可以參考下2021-10-10
這篇文章主要介紹了Python實現檢測文件MD5值的方法,涉及Python針對大文件的讀取、判斷、運算、加密等相關操作技巧,需要的朋友可以參考下2018-04-04
這篇文章主要介紹了Python抽象類應用詳情，抽象類就是控制子類的方法的名稱，要求子類必須按照父類的要求的實現指定的方法，且方法名要和父類保持一致，下文更多相關介紹需要的小伙伴可以參考一下2022-04-04
要想模型效果好，每個算法工程師都應該了解的流行超參數調優(yōu)技術。今天給大家總結超參自動優(yōu)化方法：網格搜索、隨機搜索、貝葉斯優(yōu)化?和?Hyperband，感興趣的可以了解一下2022-05-05
python opencv圖像處理(素描、懷舊、光照、流年、濾鏡 原理及實現)
這篇文章主要介紹了python 基于opencv對圖像進行各種處理，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-12-12