Pytorch中的torch.gather函數
gather函數的的官方文檔： 例子： 輸出如下： 在上面的例子中，a是一個4×4矩陣： 1）當維度dim=0，索引index_1為[3,2,1,0]時，此時可將a看成1×4的矩陣，通過index_1對a每列進行行索引：第一列第四行元素為12，第二列第三行元素為9，第三列第二行 ...
原文：pytorch 中的 gather() 函數詳解
首先，給出官方文檔的鏈接： https: pytorch.org docs stable generated torch.gather.html highlight gather torch.gather 然后，我用白話翻譯一下官方文檔。 gather，顧名思義，聚集 集合。有點像軍訓的時候，排隊一樣，把隊伍按照教官想要的順序進行排列。 還有一個更恰當的比喻：gather的作用是根據索引查找，然 ...
2021-02-15 14:43 0 700 推薦指數：
相關推薦
PyTorch中scatter和gather的用法
PyTorch中scatter和gather的用法 閑扯 許久沒有更新博客了，2019年總體上看是荒廢的，沒有做出什么東西，明年春天就要開始准備實習了，雖然不找算法崗的工作，但是還是准備在2019年的最后一個半月認真整理一下自己學習的機器學習和深度學習的知識。 scatter的用法 ...
理解pytorch幾個高級選擇函數（如gather）
目錄 1. 引言 2. 維度的理解 3. gather函數 4. index_select函數 5. masked_select函數 6. nonzero函數 1. 引言 最近在刷開源的Pytorch版動手學深度學習，里面談到幾個高級 ...
gather函數
gather(input, dim, index)：根據 index，在 dim 維度上選取數據，輸出的 size 與 index 一致 # input (Tensor) – 源張量 # dim (int) – 索引的軸 # index ...
Pytorch的gather用法理解
先放一張表，可以看成是二維數組 行(列)索引 索引0 索引1 索引2 索引3 索引0 0 ...
PyTorch分布式訓練詳解教程 scatter, gather & isend, irecv & all_reduce & DDP
PyTorch分布式訓練詳解教程 scatter, gather & isend, irecv & all_reduce & DDP 本文將從零入手，簡單介紹如何使用PyTorch中的多種方法進行分布式訓練。 具體而言，我們將使用四種方法，分別是: (1)scatter ...
pytorch函數詳解之AverageMeter
總結：本質上還是對所有batch_size的損失取平均，batch_size會在計算中被消除，並沒有啥用 reference:https://blog.csdn.net/ ...
PyTorch 中 torch.matmul() 函數的文檔詳解
官方文檔 torch.matmul() 函數幾乎可以用於所有矩陣/向量相乘的情況，其乘法規則視參與乘法的兩個張量的維度而定。 關於 PyTorch 中的其他乘法函數可以看這篇博文，有助於下面各種乘法的理解。 torch.matmul() 將兩個張量相乘划分成了五種情形：一維 × 一維 ...