Pytorch中scatter_的用法

 首先本文只介绍利用scatter方法建立one-hot向量\矩阵的用法。

 scatter()与scatter_()用法上没有不同,只不过第一个方法会返回一个新的张量,而第二个方法会在适当的位置修改原张量。

参数及含义如下:scatter(dim, index, src)

具体来说:dim指的是沿着哪个维度进行索引,若为0则代表按行索引,若为1则代表按列索引

                  index表示用来进行索引的张量

                  src表示源张量或标量,或者说需要填入的张量或标量。

x.scatter_(dim,index,src),过程是这样的:对于index按照其行或列(由dim指示)的值对应到x的那一行相应的位置上,将这个位置的0改为src。要注意的是,index的值必须要是int64,否则报错。

 例子:

import torch
x=torch.zeros(4,8)
label=torch.tensor([[1],[5],[7],[6]])
one_hot=x.scatter_(1,label,1)
print(one_hot)

 在这个例子中,one_hot=x.scatter_(1,label,1),意思就是,例如第一次修改,按照列(因为scatter_的第一个参数是1)取label中的元素,label中第一个元素是1,那么就在one_hot矩阵中的第一行将索引为1的位置,填入1(因为scatter_中第三个参数是1),以此类推。

这个例子来源于:pytorch中的scatter_()函数 - daremosiranaihana - 博客园

猜你喜欢

转载自blog.csdn.net/qq_43438974/article/details/127076760