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