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)文章
python導(dǎo)出hive數(shù)據(jù)表的schema實(shí)例代碼
這篇文章主要介紹了python導(dǎo)出hive數(shù)據(jù)表的schema實(shí)例代碼，小編覺得還是挺不錯(cuò)的，具有一定借鑒價(jià)值，需要的朋友可以參考下2018-01-01
這篇文章主要介紹了Python在字符串中處理html和xml的方法，文中講解非常細(xì)致，代碼幫助大家更好的理解和學(xué)習(xí)，感興趣的朋友可以了解下2020-07-07
python3獲取控制臺(tái)輸入的數(shù)據(jù)的具體實(shí)例
在本篇內(nèi)容里小編給大家分享的是一篇關(guān)于python3獲取控制臺(tái)輸入的數(shù)據(jù)的具體實(shí)例內(nèi)容，需要的朋友們可以學(xué)習(xí)下。2020-08-08
今天小編就為大家分享一篇關(guān)于Python并行分布式框架Celery詳解的文章，小編覺得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來(lái)看看吧2018-10-10
python實(shí)現(xiàn)自動(dòng)化上線腳本的示例
今天小編就為大家分享一篇python實(shí)現(xiàn)自動(dòng)化上線腳本的示例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-07-07
Python cookbook（數(shù)據(jù)結(jié)構(gòu)與算法）在字典中將鍵映射到多個(gè)值上的方法
這篇文章主要介紹了Python在字典中將鍵映射到多個(gè)值上的方法,涉及Python針對(duì)字典的相關(guān)映射與初始化相關(guān)操作技巧,需要的朋友可以參考下2018-02-02
詳解django的serializer序列化model幾種方法
序列化是將對(duì)象狀態(tài)轉(zhuǎn)換為可保持或傳輸?shù)母袷降倪^(guò)程。這篇文章主要介紹了詳解django的serializer序列化model幾種方法。具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-10-10
python SVM 線性分類模型的實(shí)現(xiàn)
這篇文章主要介紹了python SVM 線性分類模型的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
python3實(shí)現(xiàn)繪制二維點(diǎn)圖
今天小編就為大家分享一篇python3實(shí)現(xiàn)繪制二維點(diǎn)圖，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
python深copy和淺copy區(qū)別對(duì)比解析
這篇文章主要介紹了python深copy和淺copy區(qū)別對(duì)比解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-12-12