欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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算子用法示例的詳細內(nèi)容,更多關(guān)于pytorch算子torch.gather的資料請關(guān)注腳本之家其它相關(guān)文章!

相關(guān)文章

最新評論