這篇文章主要介紹“Python torch.gather()怎么使用”，在日常操作中，相信很多人在Python torch.gather()怎么使用問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Python torch.gather()怎么使用”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

創(chuàng)新互聯(lián)公司是一家專注于網(wǎng)站制作、網(wǎng)站建設(shè)與策劃設(shè)計,臨邑網(wǎng)站建設(shè)哪家好?創(chuàng)新互聯(lián)公司做網(wǎng)站,專注于網(wǎng)站建設(shè)十多年,網(wǎng)設(shè)計領(lǐng)域的專業(yè)建站公司;建站業(yè)務(wù)涵蓋:臨邑等地區(qū)。臨邑做網(wǎng)站價格咨詢:18980820575

使用說明:

input和index兩個參數(shù)的形狀要滿足一定的條件,
即,除了dim參數(shù)指定的軸上,其他所有的軸形狀必須完全相同.
計算得到的輸出結(jié)果out的形狀和index相同,
out輸出結(jié)果的值由input給出,具體在input中的位置由
當(dāng)前out輸出結(jié)果的位置以及index所決定,
dim軸的位置由index給出,其他軸上的位置由out的當(dāng)前輸出結(jié)果位置決定.

代碼實驗展示:

Microsoft Windows [版本 10.0.18363.1256](c) 2019 Microsoft Corporation。保留所有權(quán)利。

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>>>>> torch.manual_seed(seed=20200910)>>>>>> t = torch.tensor([[1,2],[3,4]])>>> t
tensor([[1, 2],[3, 4]])>>> torch.gather(t, 1, torch.tensor([[0,0],[1,0]]))tensor([[1, 1],[4, 3]])>>> index = torch.tensor([[0,0],[1,0]])>>> index
tensor([[0, 0],[1, 0]])>>> torch.gather(t, 1, index )tensor([[1, 1],[4, 3]])>>> torch.gather(t, 0, index)tensor([[1, 2],[3, 2]])>>>>>>>>>

到此，關(guān)于“Python torch.gather()怎么使用”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注創(chuàng)新互聯(lián)網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>
網(wǎng)站欄目：Pythontorch.gather()怎么使用
標(biāo)題路徑：http://weahome.cn/article/ijhspo.html