Pytorch torch.gather()

最新推荐文章于 2024-01-15 17:17:54 发布
最新推荐文章于 2024-01-15 17:17:54 发布
阅读量87 收藏
点赞数
分类专栏： Pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Decennie/article/details/119221642
版权
本文详细介绍了PyTorch中的torch.gather()函数，该函数用于根据指定的索引从输入tensor中收集值。文章强调了index tensor在其他维度上必须与input保持一致，且输出维度与index相同。torch.gather()常用于损失函数如CrossEntropyLoss的计算。文中通过具体的代码示例展示了如何使用该函数，并解释了其在实际问题中的应用。
摘要由CSDN通过智能技术生成
Decennie
关注
专栏目录
pytorch torch.gather 实例
Talk Is Cheap
03-08 488
import torch input_tensor = torch.tensor([[1,2],[3,4],[5,6]]) gather_input = torch.tensor([[0,0],[1,0],[1,1]]) output_tensor = torch.gather(input_tensor, 1, gather_input) print(output_tensor) tensor(...
Pytorch中的torch.gather()函数
xiaoliujun1999的博客
11-12 1446
参数说明 以官方说明为例，gather()函数需要三个参数，输入input，输出dim，以及索引index input必须为Tensor类型 dim为int类型，代表从哪个维度进行索引 index为LongTensor类型 举例说明 input=torch.tensor([[1,2,3],[4,5,6]]) #作为输入 index1=torch.tensor([[0,1,1],[0,1,1]]) #作为索引矩阵 # dim=0时,按列进行索引 print (torch.gather(i
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中torch.gather函数
12-21
在学习 CS231n中的NetworkVisualization-PyTorch任务，讲解了使用torch.gather函数，gather函数是用来根据你输入的位置索引 index，来对张量位置的数据进行合并，然后再输出。 其中 gather有两种使用方式，一种为 torch.gather 另一种为 对象.gather。 首先介绍 对象.gather import torch torch.manual_seed(2) #为CPU设置种子用于生成随机数，以使得结果是确定的 def gather_example(): N, C = 4, 5 s = torch.randn(N,
Pytorch中的torch.gather函数
believe__m的博客
09-12 505
Pytorch中的torch.gather函数，不一样的思路
PyTorch中torch.gather()函数
flyingluohaipeng的博客
11-27 2547
输入index的shape等于输出value的shape输入index的索引值仅替换该index中对应dim的index值最终输出为替换index后在原tensor中的值。
浅谈Pytorch中的torch.gather函数的含义
09-18
在PyTorch中，`torch.gather`是一个非常实用的函数，它允许用户根据提供的索引从源张量中收集值，并将这些值沿指定维度组织到一个新的张量中。这个函数在处理某些特定的数据操作时，如多分类问题中的预测结果提取，...
【PyTorch】Torch.gather()用法详细图文解释
Mocode的博客
06-05 1万+
Torch.gather()用法的详细图文解释，通过比喻理解。
[pytorch] torch.gather()函数
Myshrry的博客
07-05 503
无
Pytorch函数——torch.gather详解
最新发布
beiketaoerge的博客
01-15 1473
对pytorch中torch.gather进行了解释
pytorch的gather函数
weixin_46028606的博客
12-07 259
pytorch的gather函数
pytorch中的gather函数_Pytorch中torch.gather函数祥解
weixin_33831334的博客
02-21 565
原创申明：本文为作者原创，转载请注明出处！引言：在多分类中，torch.gather常用来取出标签所对应的概率，但对于刚开始接触Pytorch的同学来说，torch.gather()可能不太好理解，这里做一些说明和演示，帮助理解。官方说明gather( input', dim, index, out=None, sparse_grad=False)Gathers values along an a...
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函数详解，从抽象到具体
狗狗狗大王的博客
04-19 751
从官方文档出发 这个官方文档写得看起来非常不人道，但实际上它已经包含了我们理解所需要的信息，我们来仔细看一下。 首先来理解一下输入： torch.gather(input, dim, index, *, sparse_grad=False, out=None) 此处的input是一个张量，假设是个三维张量吧。 dim是指定的维度。 index也是一个张量，它必须和input有一样的维度，所以这里也假设是三维。   假设我们设置dim=1 out[i][j][k] = input[i][index[i][j
图解pytorch里面的torch.gather()
子燕若水的博客
12-07 567
上图显示了 torch gather() 函数在 dim=1 的二维张量上的工作。这里索引张量的行对应于输入张量的行（用灰色阴影突出显示）。现在对于索引张量中的每个索引值，从该行和输入张量的索引中选取相应的值。 torch.gather(input=tensor2, dim=1, index = torch.tensor([[2, 1, 0],              &nbs
【Pytorch】torch.gather用法详解
weixin_44378835的博客
05-08 2262
torch.gather torch.gather(input, dim, index, *, sparse_grad=False, out=None) 沿指定的维收集值。 参数： input (Tensor) –输入张量 dim (int) – 要索引的维 index (LongTensor) – 要收集的元素的索引 sparse_grad (bool, optional) – 如果为True，关于input 的梯度将是稀疏张量。 out (Tensor, optional) –输出
Pytorch 中 gather 函数讲解
qq_52852138的博客
04-09 1371
文章目录官方解读分析小例子 官方解读分析 该函数的功能为：沿着 dim 指定的轴收集值 torch.gather(input, dim, index, out=None) → Tensor Gathers values along an axis specified by dim. For a 3-D tensor the output is specified by: out[i][j][k] = input[index[i][j][k]][j][k] # dim=0
对于pytorch，gather()函数的理解
peacefairy的博客
09-21 202
官方文档对该函数的解释： 按自己的理解翻译的，如有错误望指出 torch.gather(input, dim, index, out=None, sparse_grad=False) → Tensor 作用：沿 dim 指定的轴收集值 参数： input (Tensor) – 要操作的张量 dim (int) – 要索引的轴 index (LongTensor) – 要收集的元素的索引 out (Tensor, optional) – 目标张量-要收集数据得到的张量 sparse_grad (bool,
torch.gather
05-01
torch.gather是PyTorch中的一个函数，用于从给定的输入Tensor中收集指定索引的值，并将结果存储在输出Tensor中。它的语法如下所示： torch.gather(input, dim, index, out=None, sparse_grad=False) 其中，输入...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值