Pytorch 学习（5）：Pytorch中的 torch.gather/scatter_ 聚集/分散操作

最新推荐文章于 2024-08-05 14:35:27 发布
最新推荐文章于 2024-08-05 14:35:27 发布
阅读量9.7k 收藏 15
点赞数 4
分类专栏： AI & Big Data案例实战课程
热烈祝贺Gavin大咖2024年北京航空航天大学两本新书《Transformer&ChatGPT解密：原理、源码及案例》、《Transformer& Rasa 解密: 原理、源码及案例》出版发行
本文链接：https://blog.csdn.net/duan_zhihua/article/details/82556676
版权
本文介绍了PyTorch中的聚集操作torch.gather和分散操作torch.scatter，详细解析了它们的工作原理和使用示例，帮助理解如何在指定轴上进行数据聚合和分布。
摘要由CSDN通过智能技术生成

 Pytorch 学习（5）：Pytorch中的 torch.gather/scatter_ 聚集/分散操作 

  • torch.gather(input, dim, index, out=None) → Tensor 聚集操作
  • torch.Tensor scatter_(dim, index, src) → Tensor 分散操作 

1.torch.gather(input, dim, index, out=None) → Tensor 聚集操作

沿给定轴 dim ,将输入索引张量 index 指定位置的值进行聚合，对一个 3 维张量,输出可以定义为:

out[i][j][k] = input[index[i][j][k]][j][k]  # if dim == 0
out[i][j][k] = input[i][index[i][j][k]][k]  # if dim == 1
out[i][j][k] = input[i][j][index[i][j][k]]  # if dim == 2

如

了解本专栏 订阅专栏 解锁全文 超级会员免费看
段智华
关注
专栏目录
订阅专栏
one hot编码：`torch.Tensor.scatter_()`函数用法详解
migue_math
02-17 3655
torch.Tensor.scatter_()是torch.gather()函数的方向反向操作。两个函数可以看成一对兄弟函数。gather用来解码one hot，scatter_用来编码one hot。 scatter_(dim, index, src) → Tensor dim (python:int) – 用来寻址的坐标轴 index (LongTensor) – 索引 src(Tensor...
Pytorch中的torch.gather函数的个人理解
CAOJUN_FANG的博客
11-21 2499
torch.gather函数的个人理解
torch.gather scatter
timertimer的博客
03-28 174
torch.gather import torch b = a.gather(dim, index) b = torch.gather(src, 1, torch.tensor([[0,0],[1,0]])) dim：指定轴方向，定义了填充方式。对于二维张量，dim=0表示逐列进行行填充，而dim=1表示逐列进行行填充。 * 当dim=1时，index[0][0]的元素是1，那么它想要查找a[0][1]中的元素; * 当dim=0时，index[0][0]的元素是1，那么它想查找的a[1][0]中的元素；
python安装torch-cluster、torch-scatter、torch-sparse和torch-geometric | torch_geometric详细安装教程
最新发布
sinat_29950703的博客
08-05 1135
python安装torch-cluster、torch-scatter、torch-sparse和torch-geometric，常见错误
torch.gather 和 torch.scatter_
m0_52305807的博客
11-14 100
刚看到这两个函数的时候不懂，到处找解说，最后发现还是官方文档说的清楚一点在此记录一下，方便自己日后回忆这是官方文档。
torch.gather与torch.scatter
chenconggan0499的博客
11-02 446
在对数据处理时，可能需要在高维张量中选取部分数据、经过处理后再“放回”原张量中。这一操作可通过torch.gather与torch.scatter实现。 torch.gather的定义为， torch.gather(input,dim,index,*,sparse_grad=False,out=None)->Tensor 以三维张量为例，输入、输出之间可表示为， out[i][j][k]=input[index[i][j][k]][j][k] # if dim == 0 out[i][j][k]=
pytorch_scatter:优化分散操作的PyTorch扩展库
05-03
该软件包包含一个高度扩展的小型扩展库，该库具有高度优化的稀疏更新（分散和分段）操作，可在使用，但主软件包中没有。 分散和分段操作可以大致描述为基于给定“组索引”张量的归约操作。 段操作要求对“组索引”...
Pytorch并行计算：nn.parallel.replicate, scatter, gather, parallel_apply
aiwanghuan5017的博客
09-17 2274
import torch import torch.nn as nn import ipdb class DataParallelModel(nn.Module): def __init__(self): super().__init__() self.block1 = nn.Linear(10, 20) def forward(...
关于pytorch中scatter_add_函数的分析、理解与实现
小卜妞的博客
10-13 4574
关于scatter_add_函数的分析、理解与实现 一、 pytorch中的定义和实现原理 在torch._C._TensorBase.py中，定义了scatter_(self, dim, index, src, reduce=None) -> Tensor方法，作用是将src的值写入index指定的self相关位置中。用一个三维张量举例如下，将src在坐标(i,j,k)下的所有值，写入self的相应位置，而self的位置坐标除了dim维度用index[i,j,k]代替以外，都不变： self[in
pytorch中scatter函数的理解
weixin_45791946的博客
03-09 344
scatter_(input, dim, index, src)：将src中数据根据index中的索引按照dim的方向填进input。可以理解成放置元素或者修改元素dim：沿着哪个维度进行索引index：用来 scatter 的元素索引src：用来 scatter 的源元素，可以是一个标量或一个张量填充规则# dim=0时，逐列（j）进行"行填充"# dim=1时，逐行（i）进行"列填充"
torch.scatter_
Mao_Jonah的博客
11-02 1645
torch.scatter_ Tensor.scatter_(dim, index, src, reduce=None) → Tensor Parameters dim (int) – the axis along which to index index (LongTensor) – the indices of elements to scatter, can be either empty or of the same dimensionality as src. When empty, the
torch_scatter
F_aF_a的博客
08-03 414
官方文档 文章目录`scatter` scatter 用一张官网的图 import torch from torch_scatter import scatter index = torch.tensor([0,0,1,0,2]) input = torch.tensor([[1,1],[1,1],[2,2],[1,1],[1,1]]) result = scatter(input,index,dim=0,reduce="sum") """ tensor([[3, 3], [2, 2],
pytorch中的gather和scatter函数
chenf1999的博客
09-12 744
最近看代码时遇到了两个函数，查阅pytorch后一时半会儿也没弄懂，现在写篇笔记来加深一下印象。 gather torch.gather(input, dim, index, out=None, sparse_grad=False) → Tensor 沿着给定的维度dim，将输入input指定位置的值聚合起来，指定位置由index决定。 index和input必须有相同数量的维度，且满足1 <= index[dim] <= input[dim]、index[other_dims] == inpu
【Torch API】关于pytorch中tensor.scatter_，torch.scatter，torch_scatter.scatter函数的分析、理解与实现
devil_son1234的博客
06-16 950
scatter摘要
pytorch中torch.Tensor.scatter用法
热门推荐
qq_39004117的博客
07-16 1万+
今天读代码，看到有人用了torch.Tensor.scatter这个函数。这个函数之前我也看到过，但是没有搞明白是干啥用的，今天我搞明白了。 首先看一下官方文档的定义：（可跳过） ==============================pytorch docs分割线==================================== scatter_(dim, index, src)...
Pytorch框架之torch.scatter_
发呆的比目鱼的博客
09-25 469
Pytorch框架之torch.scatter_
Pytorch中常见的聚合函数
weixin_43869605的博客
08-23 796
函数汇总 torch.mean() 返回平均值 torch.sum() 返回总和 torch.prod() 计算所有元素的积 torch.max() 返回最大值 torch.min() 返回最小值 torch.argmax() 返回最大值排序的索引值 torch.argmin() 返回最小值排序的索引值 ...
我想替换一个torch.nn.Module模型的forward函数中的torch.gather操作，有办法在不改变原模型代码的情况下完成吗？你能举例说明吗？
09-11
是的，你可以使用torch.index_select来替换torch.gather操作。例如，在原有模型中的代码：output = torch.gather(input, dim=1, index=indices)可以替换为：output = torch.index_select(input, dim=1, index=indices)
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

段智华

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值