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