Pytorch scatter_ 理解轴的含义

冷不防 2021-06-24 15:59 403阅读 0赞

scatter\_(input, dim, index, src)将src中数据根据index中的索引按照dim的方向填进input中。

>>> x = torch.rand(2, 5)
    >>> x
    
     0.4319  0.6500  0.4080  0.8760  0.2355
     0.2609  0.4711  0.8486  0.8573  0.1029
    [torch.FloatTensor of size 2x5]

LongTensor的shape刚好与x的shape对应，也就是LongTensor每个index指定x中一个数据的填充位置。dim=0，表示按行填充，主要理解按行填充。举例LongTensor中的第0行第2列index=2，表示在第2行（从0开始）进行填充填充，对应到zeros(3, 5)中就是位置（2,2）。所以此处要求zeros(3, 5)的列数要与x列数相同，而LongTensor中的index最大值应与zeros(3, 5)行数相一致。

>>> torch.zeros(3, 5).scatter_(0, torch.LongTensor([[0, 1, 2, 0, 0], [2, 0, 0, 1, 2]]), x)
    
     0.4319  0.4711  0.8486  0.8760  0.2355
     0.0000  0.6500  0.0000  0.8573  0.0000
     0.2609  0.0000  0.4080  0.0000  0.1029
    [torch.FloatTensor of size 3x5]

同上理，可以把1.23看成\[\[1.23\], \[1.23\]\]。此处按列填充，LongTensor中的index=2对应zeros(2, 4)的(0,2)位置。

>>> z = torch.zeros(2, 4).scatter_(1, torch.LongTensor([[2], [3]]), 1.23)
    >>> z
    
     0.0000  0.0000  1.2300  0.0000
     0.0000  0.0000  0.0000  1.2300
    [torch.FloatTensor of size 2x4]

发表评论取消回复

表情：

评论列表（有 0 条评论，403人围观）

还没有评论，来说两句吧...

相关阅读

相关 Pytorch plt.scatter()函数用法

一.scatter（）函数的定义 matplotlib.pyplot.scatter(x, y, s=None, c=None, marker=None, cmap=No

傷城~/ 2024年02月06日 06:36/ 0 赞/ 74 阅读

相关理解getTypeParameters的含义

使用java泛型有时会看到getTypeParameters方法，这个方法是什么意思，下面就来具体了解一下：官方给出的解释如下： Returns an array of

た入场券/ 2023年10月09日 20:31/ 0 赞/ 28 阅读

相关 Pytorch：使用scatter()、scatter_()将Label变成one-hot编码

TORCH.TENSOR.SCATTER\_ Tensor.scatter\_(dim, index, src, reduce=None) → [Tensor][] W

梦里梦外;/ 2023年09月25日 14:33/ 0 赞/ 41 阅读

相关 Pytorch：torch.Tensor.scatter()和torch.Tensor.gather

[miguemath][]指出：torch.Tensor.scatter\_()是torch.gather()函数的方向反向操作。两个函数可以看成一对兄弟函数。gather用来

逃离我推掉我的手/ 2023年01月08日 08:25/ 0 赞/ 177 阅读

相关 loU含义的理解

IoU是两个区域重叠的部分除以两个区域的集合部分得出的结果，通过设定的阈值，与这个IoU计算结果比较。 ![watermark_type_ZmFuZ3poZW5naGVpdG

ゝ一纸荒年。/ 2023年01月03日 12:49/ 0 赞/ 189 阅读

相关 mAP含义的理解

在目标检测中，Precision就是检测出来的框的数目（或者是面积吧），除以一共画出来的框。比如说有时候你预测出来10个框，只有5个是正确的，那Precision就是5/10

心已赠人/ 2023年01月03日 12:49/ 0 赞/ 185 阅读

相关如何理解python中的0轴和1轴

最近在学习的过程中碰到一个之前很熟悉的知识点，现在却傻傻分不清，就是我们在看代码的时候经常会看到某个函数里面 F.log\_softmax(y / temp, dim=1) ，

朴灿烈づ我的快乐病毒、/ 2022年11月09日 01:00/ 0 赞/ 163 阅读

相关 pandas的scatter_matrix散布矩阵图的理解

![这里写图片描述][SouthEast] Q: 如何理解问题3中给出的图？如何分析关联性、变量分布？ A: 这张图分为两部分：对角线部分和非对角线部分。对

太过爱你忘了你带给我的痛/ 2022年05月12日 08:24/ 0 赞/ 186 阅读

相关 Pytorch scatter_ 理解轴的含义

scatter\_(input, dim, index, src)将src中数据根据index中的索引按照dim的方向填进input中。 >>> x = torch.

冷不防/ 2021年06月24日 15:59/ 0 赞/ 404 阅读

相关 pandas 中的Axis（轴）含义

![这里写图片描述][SouthEast] 根据stackoverflow答主解释，axis=0指的是逐行，axis=1指的是逐列。 df = pd.DataFr

蔚落/ 2021年06月10日 20:42/ 0 赞/ 438 阅读