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算子用法示例的詳細內(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),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07Python中的pathlib.Path為什么不繼承str詳解
這篇文章主要給大家介紹了關(guān)于Python中pathlib.Path為什么不繼承str的相關(guān)資料,文中通過示例代碼介紹的非常詳細,對大家學(xué)習(xí)或者使用Python具有一定的參考學(xué)習(xí)價值,需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06Python requests模塊基礎(chǔ)使用方法實例及高級應(yīng)用(自動登陸,抓取網(wǎng)頁源碼)實例詳解
這篇文章主要介紹了Python requests模塊基礎(chǔ)使用方法實例及高級應(yīng)用(自動登陸,抓取網(wǎng)頁源碼,Cookies)實例詳解,需要的朋友可以參考下2020-02-02