pytorch gather

最新推荐文章于 2024-04-08 20:16:14 发布

开心就哈哈

最新推荐文章于 2024-04-08 20:16:14 发布

阅读量416

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/pain_gain0/article/details/100998152

版权

pytorch 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

y
Out[34]: tensor([0, 2])
y_hat
Out[35]: 
tensor([[0.1000, 0.3000, 0.6000],
        [0.3000, 0.2000, 0.5000]])

y_hat.gather(1, y.view(-1, 1))

聚合方向y_hat的维度1，聚合位置：

a[0][y.view(-1,1)[0]] = 0.1

a[1][y.view(-1,1)[1]] = 0.5

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

开心就哈哈

关注关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

pytorch torch.gather 实例

Talk Is Cheap

03-08

484

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()函数的用法及在Focal Loss中的应用(详细易懂)

热门推荐

畅游计算机视觉的海洋

05-25

19万+

文章目录官方文档中torch.gather的用法torch.gather应用总结官方文档中torch.gather的用法 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]

参与评论您还未登录，请先登录后发表或查看评论

【PyTorch】Torch.gather()用法详细图文解释

Mocode的博客

06-05

1万+

Torch.gather()用法的详细图文解释，通过比喻理解。

pytorch中的gather函数

Abraham Ben

10-10

3007

pytorch中的gather函数比较难以理解，下面用2D的例子解释如下： import torch input = torch.arange(16).view(4, 4) """ [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]] """ 首先要指出的是input和index必须有相同的维度，但是各个维度的大小(shape)可以不一样。比如input是一个三维矩阵，则tensor也必须是一个三位矩阵。对于d != dim的维度

pytorch中gather用法

weixin_49327481的博客

03-06

5799

pytorch中gather用法

pytorch的gather

sdhdsf132452的博客

04-03

1966

1. 官网解释 out[i][j][k]=input[[index[i][j][k]][j][k], if dim==0out[i][j][k] = input[[index[i][j][k]][j][k],\ if \ dim==0out[i][j][k]=input[[index[i][j][k]][j][k], if dim==0 以此类推意思就是说，假设我要求out[i][j]out[i][j]out[i][j]位置的值，如果dim=0，那么i就会被inde

Pytorch的gather()和scatter()

01-21

Pytorch的gather()和scatter() 1.gather() gather是取的意思，意为把某一tensor矩阵按照一个索引序列index取出，组成一个新的矩阵。 gather(input,dim,index) 参数： input是要取值的矩阵 dim指操作的维度，0为竖向...

Pytorch 的 gather的直观理解(做个记录)

fengyanghe的专栏

03-03

487

torch.gather(input, dim, index, out=None) 直观的通俗解释: 目的是从input中,选择index中所指示位置的数值,然后生成一个新的tensor. 再直观的说: 目的是生成一个tensor, 这个新的tensor里的数值都是来自input的, 到底tensor中的哪个位置用input的哪个元素,是由index和dim共同确定的. 这里面有两个...

pytorch gather，index_fill函数

qq_40728805的博客

12-13

4525

pytorch: gather函数，index_fill函数 torch.gather(input, dim, index, out=None) → Tensor In [28]: a=torch.arange(0,16).view(4,4) In [29]: a Out[29]: tensor([[ 0, 1, 2, 3], [ 4, 5, 6, 7], ...

Pytorch高阶OP操作where，gather原理

01-21

PyTorch是一个非常有可能改变深度学习领域前景的Python库。我尝试使用了几星期PyTorch...二、gather 1）官方解释：根据指定的维度和索引值来筛选值 2）举例您可能感兴趣的文章:浅谈Pytorch torch.optim优化器个性化

深入理解PyTorch中的gather函数

m0_48319758的博客

04-13

699

gather函数今天在用PyTorch复现softmax的时候，参考的书籍为《Dive into DL Pytorch》。在书里面，关于gather函数原文是这样叙述的：上一节中，我们介绍了softmax回归使用的交叉熵损失函数。为了得到标签的预测概率，我们可以使用gather函数。在下面的例子中，变量y_hat是2个样本在3个类别的预测概率，变量y是这2个样本的标签类别。通过使用gather函数，我们得到了2个样本的标签的预测概率。在代码中，标签类别的离散值是从0开始逐一递增的。 y_hat =

Pytorch中的gather()

chen_znn的博客

07-05

1236

torch.gather()使用记录

【pytorch函数】gather函数

最新发布

qq_44091004的博客

04-08

402

输入张量是一个2行2列的张量，0表示以列为单位进行索引，index_tensor表示对应列上的索引值，比如在第0列上取1,1位置上的值填充到原来的位置，第1列上取1,1位置上的值填充到原来的位置，也就是说只能在各自列或者各自行上的相应位置的元素替换为该列或者该行任意位置的元素。

pytorch.gather()函数

weixin_48319333的博客

01-11

130

【代码】pytorch.gather()函数。

pytorch的gather函数

qq_40589781的博客

02-08

230

官方文档 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]][...

pytorch gather函数

我的博客

12-22

866

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, d...

pytorch中的gather函数_Pytorch中torch.gather函数祥解

weixin_33831334的博客

02-21

556

原创申明：本文为作者原创，转载请注明出处！引言：在多分类中，torch.gather常用来取出标签所对应的概率，但对于刚开始接触Pytorch的同学来说，torch.gather()可能不太好理解，这里做一些说明和演示，帮助理解。官方说明gather( input', dim, index, out=None, sparse_grad=False)Gathers values along an a...

PyTorch典型函数之gather

摇光65535的博客

05-10

649

介绍了PyTorch典型函数gather的使用场景与用法。

pytorch 里的gather

07-15

在PyTorch中，`gather`函数用于根据索引从输入张量中收集元素。它的使用方式如下： ```python torch.gather(input, dim, index, out=None) ``` 其中，`input`是输入张量，`dim`是指定收集的维度，`index`是包含索引的张量。具体而言，`gather`函数会根据索引张量 `index` 在指定维度 `dim` 上收集 `input` 张量中的元素。最后，它会返回一个新的张量，其形状与 `index` 相同。以下是一个示例： ```python import torch # 创建输入张量 input = torch.tensor([[1, 2], [3, 4], [5, 6]]) # 创建索引张量 index = torch.tensor([[0, 1], [1, 0]]) # 在第1维度上收集元素 output = torch.gather(input, 0, index) print(output) ``` 输出结果为： ``` tensor([[1, 4], [3, 2]]) ``` 在上述示例中，输入张量 `input` 是一个2维张量，索引张量 `index` 也是一个2维张量。通过指定维度0，我们从 `input` 中收集了 `index` 中对应的元素，得到了输出张量 `output`。希望这可以帮助到你！如果有任何疑问，请随时提问。