原文:Pytorch中的torch.gather函數

gather函數的的官方文檔: 例子: 輸出如下: 在上面的例子中,a是一個 矩陣: 當維度dim ,索引index 為 , , , 時,此時可將a看成 的矩陣,通過index 對a每列進行行索引:第一列第四行元素為 ,第二列第三行元素為 ,第三列第二行元素為 ,第四列第一行元素為 ,即b , , , 當維度dim ,索引index 為 , , , T時,此時可將a看成 的矩陣,通過index 對 ...

2020-08-30 15:48 0 633 推薦指數:

查看詳情

pytorch gather() 函數詳解

首先,給出官方文檔的鏈接: https://pytorch.org/docs/stable/generated/torch.gather.html?highlight=gather#torch.gather 然后,我用白話翻譯一下官方文檔。 gather,顧名思義,聚集 ...

Mon Feb 15 22:43:00 CST 2021 0 700
Pytorchtorch.where函數

首先我們看一下Pytorchtorch.where函數是怎樣定義的: torch.where函數的功能如下: 以具體實例看一下torch.where函數的效果: 結果如下: 可以看到torch.where函數會對condition ...

Wed Dec 29 23:17:00 CST 2021 0 1044
pytorchtorch.narrow()函數

torch.narrow(input, dim, start, length) → Tensor Returns a new tensor that is a narrowed version of input tensor. The dimension dim is input from ...

Fri Nov 15 05:18:00 CST 2019 0 1238
Pytorchtorch.cat()函數

以把list的tensor拼接起來。 比如: 上面的代碼可以合成一行來寫: ...

Mon Jul 15 23:03:00 CST 2019 0 1921
Pytorchtorch.cat()函數

cat是concatnate的意思:拼接,聯系在一起。 先說cat( )的普通用法 如果我們有兩個tensor是A和B,想把他們拼接在一起,需要如下操作: 其次,cat還可以把list的tensor拼接起來。 比如: 上面的代碼可以合成 ...

Sun Dec 23 04:29:00 CST 2018 0 43400
PyTorchscatter和gather的用法

PyTorchscatter和gather的用法 閑扯 許久沒有更新博客了,2019年總體上看是荒廢的,沒有做出什么東西,明年春天就要開始准備實習了,雖然不找算法崗的工作,但是還是准備在2019年的最后一個半月認真整理一下自己學習的機器學習和深度學習的知識。 scatter的用法 ...

Sun Nov 17 19:21:00 CST 2019 0 457
pytorchtorch.cat(),torch.chunk(),torch.split()函數的使用方法

一、torch.cat()函數 熟悉C字符串的同學們應該都用過strcat()函數,這個函數在C/C++程序中用於連接2個C字符串。在pytorch,同樣有這樣的函數,那就是torch.cat()函數. 先上源碼定義:torch.cat(tensors,dim=0,out=None ...

Mon Apr 13 00:09:00 CST 2020 2 6877
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM