标签:11 10 12 函数 gather PyTorch tensor
整理自官方文档:https://www.oschina.net/action/GoToLink?url=https%3A%2F%2Fpytorch.org%2Fdocs%2Fstable%2Fgenerated%2Ftorch.gather.html%3Fhighlight%3Dgather%23torch.gather
0x01 背景
在做强化学习DDQN实(CtrlC)验(CtrlV)的时候遇到了一些不是很理解的函数,这里做一些解读。
0x02 gather函数
gather,直译为聚合,聚集。
先跑两个例子示范一下它会出来什么东西:
首先建立两个tensor:a和b
a = torch.tensor([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]])
tensor([[1, 2, 0, 0, 0], [0, 1, 0, 0, 0], [0, 0, 0, 1, 0]])
下面展示一下效果:
>>> a
tensor([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[11, 12, 13, 14, 15]])
>>> b
tensor([[1, 2, 0, 0, 0],
[0, 1, 0, 0, 0],
[0, 0, 0, 1, 0]])
>>> a.gather(0, b)
tensor([[ 6, 12, 3, 4, 5],
[ 1, 7, 3, 4, 5],
[ 1, 2, 3, 9, 5]])
>>> a.gather(1, b)
tensor([[ 2, 3, 1, 1, 1],
[ 6, 7, 6, 6, 6],
[11, 11, 11, 12, 11]])
gather函数其实是就相当于是对原来的tensor做一个选取和替换。
第一个参数是dim,表明我们要在哪个维度上做选择(比如是在矩阵的行还是列上做选择)。
第二个参数是索引,不一定非得和原来的a的形状保持一致。
下面,我们直接用一张网图来说明gather函数的工作原理。
(本来打了很多字,但是还是说不清楚QAQ
标签:11,10,12,函数,gather,PyTorch,tensor 来源: https://blog.csdn.net/weixin_43466027/article/details/117385716
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。