where() 條件判斷 gather()查表 input ：待查的表 dim ： 在input的哪個維度上查表 index： 待查表的索引值 ...

1.Broadcasting Broadcasting能夠實現Tensor自動維度增加（unsqueeze）與維度擴展（expand），以使兩個Tensor的shape一致，從而完成某些操作，主要按照如下步驟進行： 從最后面的維度開始匹配（一般后面理解為小維度）； 在前面插入若干維度 ...

一、Pytorch安裝 安裝cuda和cudnn，例如cuda10，cudnn7.5 官網下載torch：https://pytorch.org/ 選擇下載相應版本的torch 和torchvision的whl文件 使用pip install whl_dir安裝torch，並且同時安裝 ...

先放一張表，可以看成是二維數組 行(列)索引 索引0 索引1 索引2 索引3 索引0 0 ...

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

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

看得不明不白（我在下一篇中寫了如何理解gather的用法） gather是一個比較復雜的操作，對一個2維tensor，輸出的每個元素如下： 二維tensor的gather操作 針對0軸 注意index此時的值 輸入 輸出 分割線============ 針對1軸 ...

gather函數的的官方文檔： 例子： 輸出如下： 在上面的例子中，a是一個4×4矩陣： 1）當維度dim=0，索引index_1為[3,2,1,0]時，此時可將a看成1×4的矩陣，通過index_1對a每列進行行索引：第一列第四行元素為12，第二列第三行元素為9，第三列第二行 ...