pytorch之gather()

最新推荐文章于 2022-07-30 15:43:03 发布
最新推荐文章于 2022-07-30 15:43:03 发布
阅读量123 收藏
点赞数
分类专栏： RL torch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/fxolivia/article/details/104609703
版权
RL 同时被 2 个专栏收录
4 篇文章 0 订阅
订阅专栏
torch
2 篇文章 0 订阅
订阅专栏 
>>>import torch
>>>
>>>a = torch.Tensor([[1,2,3],[4,5,6]])
>>>a
tensor([[1., 2., 3.],
        [4., 5., 6.]])
>>>b = torch.gather(a,1,torch.LongTensor([[1,1,1],[0,0,0]]))
>>>b
tensor([[2., 2., 2.],
        [4., 4., 4.]])
>>>b = torch.gather(a,1,torch.LongTensor([[0,0,2],[1,1,0]]))
>>>b
tensor([[1., 1., 3.],
        [5., 5., 4.]])
fxolivia
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Pytorch的gather()和scatter()
01-21
Pytorch的gather()和scatter() 1.gather() gather是取的意思，意为把某一tensor矩阵按照一个索引序列index取出，组成一个新的矩阵。 gather(input,dim,index) 参数： input是要取值的矩阵 dim指操作的维度，0为竖向...
Pytorch高阶OP操作where，gather原理
09-16
主要介绍了Pytorch高阶OP操作where，gather原理,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
torch.gather() 和torch.sactter_()的用法简析
朝花&夕拾
08-29 1万+
  torch.gather(input, dim, index, out=None)  和 torch.scatter_(dim, index, src)是一对作用相反的方法   先来看torch.gather， 核心操作其实就是这样： out[i][j][k] = input[index[i][j][k]] [j][k]  # if dim == 0 out[i][j][k] =...
pytorch中gather()方法
ZYripe的博客
04-02 1727
1 代码示例： import torch a = torch.arange(15).view(3, 5) b = torch.zeros_like(a) b[1][2] = 1 b[0][0] = 2 a、b矩阵分别为： a矩阵 tensor([[ 0, 1, 2, 3, 4], [ 5, 6, 7, 8, 9], [10, 11, 12, 13, 14]]) b矩阵 tensor([[2, 0, 0, 0, 0], [0, 0, 1, 0
pytorch中的gather函数
gsgbgxp的博客
02-14 7526
在学习强化学习的过程中，有时需要将 理解pytorch几个高级选择函数（如gather） https://www.cnblogs.com/liangjianli/p/13754817.html#3-gather%E5%87%BD%E6%95%B0 gather函数 pytorch和numpy中许多函数都涉及维度运算，gather也不例外，但是它相对于其他函数更难理解。依然先来看一个例子 import torch a = torch.arange(1, 16).reshape(5, 3) “”" result
Pytorch系列（1）：torch.gather()
cpluss的博客
05-16 4万+
torch.gather 作用：收集输入的特定维度指定位置的数值 参数： input(tensor):   待操作数。不妨设其维度为（x1, x2, …, xn） dim(int):   待操作的维度。 index(LongTensor):   如何对input进行操作。其维度有限定，例如当dim=i时，index的维度为（x1, x2, …y, …,xn），...
pytorch中的gather()函数
木盏
09-22 1万+
首先，给出官方文档的链接： https://pytorch.org/docs/stable/generated/torch.gather.html?highlight=gather#torch.gather 然后，我用白话翻译一下官方文档。 gather，顾名思义，聚集、集合。有点像军训的时候，排队一样，把队伍按照教官想要的顺序进行排列。 还有一个更恰当的比喻：gather的作用是根据索引查找，然后讲查找结果以张量矩阵的形式返回。 1. 拿到一个张量： import torch a = tor
浅谈Pytorch中的torch.gather函数的含义
09-18
今天小编就为大家分享一篇浅谈Pytorch中的torch.gather函数的含义，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch中torch.gather函数
12-21
在学习 CS231n中的NetworkVisualization-PyTorch任务，讲解了使用torch.gather函数，gather函数是用来根据你输入的位置索引 index，来对张量位置的数据进行合并，然后再输出。 其中 gather有两种使用方式，一种为 ...
pytorch中gather函数详解【包你看懂，我敢说在CSDN上没人解释的比我清楚】
qq_23022733的博客
07-30 1379
在看centernet中遇到gather函数，看了好几个博客也都没咋看懂，直到看了这个视频。上面的代码中，c、d为output，a为input，b为index，首先解释一下什么是dim=0和dim=1，其实0和1是。索引的位置，例如dim=0，以3维为例，，在此感谢这位哔站up主。.........
pytorch中gather用法
weixin_49327481的博客
03-06 5719
pytorch中gather用法
pytorch中的gather函数_理解pytorch几个高级选择函数（如gather）
weixin_39682944的博客
02-21 639
1. 引言最近在刷开源的Pytorch版动手学深度学习，里面谈到几个高级选择函数，如index_select，masked_select，gather等。这些函数大多很容易理解，但是对于gather函数，确实有些难理解，官方文档开始也看得一脸懵，感觉不太直观。下面谈谈我对这几个函数的一些理解。2. 维度的理解对于numpy和pytorch，其数组在做维度运算上刚开始可能会给人一种直观上的误解，以n...
pytorch之gather()方法
Lavi的专栏
11-30 1万+
首先，先给出torch.gather函数的函数定义： torch.gather(input, dim, index, out=None) → Tensor 官方给出的解释是这样的： 沿给定轴dim，将输入索引张量index指定位置的值进行聚合。 对一个3维张量，输出可以定义为： out[i][j][k] = tensor[index[i][j][k]][j][k] # dim=0 out[i][j...
Pytorch中的torch.gather函数的含义
热门推荐
兔角与禅
05-30 7万+
pytorch中的gather函数 pytorch比tensorflow更加编程友好，所以准备用pytorch试着做最近要做的一些实验。 立个flag开始学习pytorch，新开一个分类整理学习pytorch中的一些踩到的泥坑。 今天刚开始接触，读了一下documentation，写一个一开始每太搞懂的函数gather b = torch.Tensor([[1,2,3],[4,5,6]...
Laravel 学习笔记： 授权策略（Policy）
她和她的猫的博客
07-03 4770
Policy（即策略）是在特定模型或者资源中组织授权逻辑的类，用来处理用户授权动作。
torch.gather()函数
Apikaqiu的博客
02-10 1万+
b = torch.Tensor([[1,2,3],[4,5,6]]) print(b) index_1 = torch.LongTensor([[0,1],[2,0]]) index_2 = torch.LongTensor([[0,1,1],[0,0,0]]) print (torch.gather(b, dim=1, index=index_1)) print (torch.gather(b...
pytorch.gather一点理解,为啥要造这个玩意
qq_27628577的博客
12-03 188
# 首先有一个input,取他的元素按照index来, # e.g # a=[[1,2],[3,4]] ----取他的元素----> a[行1][列1] ,..., a[行n][列n] # 记作 index = [[行],[列]] # 即阔一发现 index 维度是和 a的shape一样 # 回到问题,作为一个noob,我费解的这样取索引不就好了,那么大神复杂设计背后 # 除了显...
Pytorch之gather的用法（易懂的方式解释）
ACMer
06-20 4108
百度了一圈gather的用法，看的一知半解，结合了几篇博客的讲解，终于理解了这个的用法，记录下来，用于以后忘记的时候自己可以快速复习，同时不懂得小伙伴也可以参考下我这得理解，或许能帮助到你！！！ 首先了解下函数中的参数： torch.gather(input, dim, index, out=None) → Tensor Parameters: input (Tensor) – The source tensor dim (int) – The axis along wh
pytorch的gather()方法详解
qq_30653631的博客
12-29 2176
首先，先将结果展示出来，后续根据结果来进行分析 t = torch.Tensor([[1,2,3],[4,5,6]]) index_a = torch.LongTensor([[0,0],[0,1]]) index_b = torch.LongTensor([[0,1,1],[1,0,0]]) print(t) print(torch.gather(t,dim=1,index=index_a)) print(torch.gather(t,dim=0,index=index_b)) 运行结果如下： >
pytorch 里的gather
最新发布
07-15
在PyTorch中，`gather`函数用于根据索引从输入张量中收集元素。它的使用方式如下： ```python torch.gather(input, dim, index, out=None) ``` 其中，`input`是输入张量，`dim`是指定收集的维度，`index`是包含...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值