Pytorch---之scatter_ 理解轴的含义

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]

转载：https://blog.csdn.net/qq_16234613/article/details/79827006

Pytorch---之scatter_ 理解轴的含义相关推荐

Pytorch——torch.Tensor.data操作的含义
1 致谢感谢陈助教的帮助! 2 前言今天在学习FaceBoxes~ 看到一句代码不是很懂, prior_data = priors.data 因为data是torch.Tensor类的一个成员变量 ...
lstm 输入数据维度_理解Pytorch中LSTM的输入输出参数含义
本文不会介绍LSTM的原理,具体可看如下两篇文章 Understanding LSTM Networks DeepLearning.ai学习笔记(五)序列模型 -- week1 循环序列模型 1.举个 ...
pytorch：scatter_
我终于想明白这个函数的意思啦!!! scatter的意思:服务组件架构 data.scatter_(dim,index,src) 将src中数据根据index中的索引按照dim的方向填进data. i ...
动手学深度学习第一次打卡
今年才开始接触深度学习,基本上是小白的状态,希望能利用疫情放假的这些日子好好学习一下深度学习.第一次打卡包括Task01和Task02两部分,Task01的学习任务又分为线性回归.Softmax与分类 ...
NumPy简明教程（二、数组2）
NumPy数组(2.数组的操作) 基本运算数组的算术运算是按元素逐个运算.数组运算后将创建包含运算结果的新数组. [python] view plain copy >>> a= n ...
IDE Framework之PyTorch使用记录
文章目录 PyTorch中的常用操作序号001 torchTensor.item() A.numel()=a, B.numel()=b, 已知a>b, 从A中随机采样b个元素赋值给B, 方法是 ...
Pytorch的LSTM的理解
20211227 lstm和gru的区别 Pytorch实现LSTM案例学习(1)_ch206265的博客-CSDN博客_pytorch搭建lstm lstm案例 class torch.nn.LST ...
pytorch reshape_PyTorch中的contiguous
本文讲解了pytorch中contiguous的含义.定义.实现,以及contiguous存在的原因,非contiguous时的解决办法.并对比了numpy中的contiguous. contiguo ...
【PyTorch】contiguous==＞保证Tensor是连续的，通常transpose、permute 操作后执行 view需要此方法
目录 PyTorch中的is_contiguous是什么含义? 行优先为什么需要 contiguous ? contiguous 本身是形容词,表示连续的,关于 contiguous, PyTorc ...

Pytorch---之scatter_ 理解轴的含义

Pytorch---之scatter_ 理解轴的含义相关推荐

最新文章

热门文章