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() 将两个张量相乘划分成了五种情形：一维 × 一维 ...