pytorch人工智能之torch.gather算子用法示例
更新時間：2022年09月19日 16:09:07 作者：ctrlA_ctrlC_ctrlV
這篇文章主要介紹了pytorch人工智能之torch.gather算子用法示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
一、用法：
torch.gather 算子用于返回給定索引/下標(biāo)的 Tensor 元素，在 pytorch 官網(wǎng)文檔中的定義如下：
torch.gather( input, dim, index, *, sparse_grad=False, out=None) → Tensor
其用法等價于：
input.gather( dim, index, *, sparse_grad=False, out=None) → Tensor
其中，input 是目標(biāo) Tensor ，即被搜索的 Tensor ；dim 是搜索維度（也是 Tensor ），index 是索引。
返回值類型：Tensor
二、代碼示例：
概念看不懂沒關(guān)系，一看代碼便知用法。
a = torch.tensor([1, 5, 3, 6, 8]) b = torch.tensor([3]) # 索引為3 c = a.gather(0, b) # 輸出a中第0維索引是3的元素：6 # 等價于 c=torch.gather(a,0,b) print(c) # tensor([6])
a = torch.tensor([[1.3, 2, 3, 4.5, 5],
[2.0, 3, 0.3, 4.1, 2],
[6, 7, 8, 9, 2],
[10, 5, 0, 6, 8]])
b = torch.tensor([[1],
[2],
[3],
[4]])
c = torch.gather(a, 1, b) # 輸出a中第1維索引分別是1，2，3，4的元素：2，0.3，9，8
print(c) # tensor([[2.0000],[0.3000],[9.0000],[8.0000]])以上就是pytorch人工智能之torch.gather算子用法示例的詳細(xì)內(nèi)容，更多關(guān)于pytorch算子torch.gather的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
Pytorch數(shù)據(jù)類型Tensor張量操作的實現(xiàn)
本文主要介紹了Pytorch數(shù)據(jù)類型Tensor張量操作的實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07
Python中的pathlib.Path為什么不繼承str詳解
這篇文章主要給大家介紹了關(guān)于Python中pathlib.Path為什么不繼承str的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家學(xué)習(xí)或者使用Python具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06
今天小編給大家介紹一款可視化模塊，使用它可以繪制出十分驚艷的動圖效果。文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2022-04-04
pandas對象擁有一組常用的數(shù)學(xué)和統(tǒng)計方法,大部分都屬于約簡和匯總統(tǒng)計,這篇文章主要給大家介紹了關(guān)于pandas求平均數(shù)和中位數(shù)的相關(guān)資料,需要的朋友可以參考下2021-08-08
這篇文章主要介紹了python制作一個桌面便簽軟件分別給大家附上ubuntu和windows版的程序及源碼，有需要的小伙伴可以參考下。2015-08-08
這篇文章主要介紹了詳解Python在使用JSON時需要注意的編碼問題，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-12-12
這篇文章主要介紹了教你利用Python+Turtle繪制簡易版愛心表白,文中有非常詳細(xì)的代碼示例,對想要和男朋友或者女朋友表白的小伙伴們有很大幫助喲,需要的朋友可以參考下2021-04-04
這篇文章主要介紹了python實現(xiàn)簡單ftp客戶端的方法,涉及ftplib模塊的相關(guān)使用技巧,需要的朋友可以參考下2015-06-06
這篇文章主要介紹了Python實現(xiàn)微信消息防撤回 ,需要的朋友可以參考下2019-04-04
Python requests模塊基礎(chǔ)使用方法實例及高級應(yīng)用(自動登陸,抓取網(wǎng)頁源碼)實例詳解
這篇文章主要介紹了Python requests模塊基礎(chǔ)使用方法實例及高級應(yīng)用(自動登陸,抓取網(wǎng)頁源碼,Cookies)實例詳解,需要的朋友可以參考下2020-02-02