pytorch 笔记：gather 函数

torch.gather(input, dim, index, out=None) → Tensor

我们直接用例子来说明好了

以三位张量为例

out[i] [j] [k] = tensor[index[i][j][k]]  [j]  [k]
# dim=0out[i] [j] [k] = tensor[i]  [index[i][j][k]]  [k]
# dim=1out[i] [j] [k] = tensor[i]  [j]  [index[i][j][k]]
# dim=3

import torch
a = torch.Tensor([[1,2],[3,4]])
torch.gather(a,0,index=torch.LongTensor([[0,0],[1,0]]))
'''
tensor([[1., 2.],[3., 2.]])
'''

这个怎么看呢

out[0][0]	a[index[0][0]] [0]]	a[0][0]=1
out[1][0]	a[index[1][0]] [0]]	a[1][0] =3
out[0][1]	a[index[0][1]] [1]]	a[0][1]=2
out[1][1]	a[index[1][1]] [1]]	a[0][1]=2

还有一种用法在每一行选择一个的时候比较常用

import torch
a = torch.Tensor([[1,2,5,6],[3,4,7,8]])
torch.gather(a,1,torch.LongTensor([[0],[3]]))
'''
tensor([[1.],[8.]])
'''

pytorch 笔记：gather 函数相关推荐

Pytorch中gather函数的个人理解方法
之前一直理解不了Pytorch中gather的用法,看了官方的文档也是一头雾水.然后自己琢磨,找规律,用以下方法进行理解. 一.官方文档 torch.gather(input, dim, index, ...
PyTorch中gather()函数的用法
torch.gather(input, dim, index, out=None) → Tensor 沿给定轴,按照索引张量将原张量的指定位置的元素重新聚合成一个新的张量参数含义: input (T ...
Pytorch 中 gather 函数讲解
文章目录官方解读分析小例子官方解读分析该函数的功能为:沿着 dim 指定的轴收集值 torch.gather(input, dim, index, out=None) → TensorGath ...
pytorch中gather函数的理解
官方解释,很清楚了 torch.gather(input,dim,index,out=None) → Tensortorch.gather(input, dim, index, out=None) → ...
torch的拼接函数_从零开始深度学习Pytorch笔记（13）—— torch.optim
前文传送门: 从零开始深度学习Pytorch笔记(1)--安装Pytorch 从零开始深度学习Pytorch笔记(2)--张量的创建(上) 从零开始深度学习Pytorch笔记(3)--张量的创建(下) ...
gather torch_浅谈Pytorch中的torch.gather函数的含义
pytorch中的gather函数 pytorch比tensorflow更加编程友好,所以准备用pytorch试着做最近要做的一些实验. 立个flag开始学习pytorch,新开一个分类整理学习pyt ...
Pytorch的使用：torch.gather函数
Pytorch的使用:torch.gather函数 **torch.gather()** 作用:方便从批量tensor中获取特定化维度指定索引下的数据,该索引往往是乱序的. 首先看一下官方文档中的3维 ...
PyTorch入门笔记-matmul函数详解
PyTorch入门笔记-matmul函数详解本文转载自:PyTorch入门笔记-matmul函数详解 - 腾讯云开发者社区-腾讯云 (tencent.com) 41409)]
Pytorch中的torch.gather函数的含义
pytorch中的gather函数 pytorch比tensorflow更加编程友好,所以准备用pytorch试着做最近要做的一些实验. 立个flag开始学习pytorch,新开一个分类整理学习pyt ...
pytorch中的gather函数_理解pytorch几个高级选择函数（如gather）
1. 引言最近在刷开源的Pytorch版动手学深度学习,里面谈到几个高级选择函数,如index_select,masked_select,gather等.这些函数大多很容易理解,但是对于gather ...

pytorch 笔记：gather 函数

pytorch 笔记：gather 函数相关推荐

最新文章

热门文章