pytorch index_add_用法介绍

x.index_add_(0, assi, y)

代表将y中的每一个向量,分别加在x的不同向量上,具体加在x的哪一个向量上看assi。

x是被加数,y是加数,结果是和。该函数的三个参数分别是:

第一个参数0代表相加的维度,

第二个参数assi代表相加的项目,

第三个参数y代表加数。

如下图,以全0的x为例子,在dim=0上用y加在x上。

 

该函数中参数assi的最外一个维度和y的最外一个维度必须保持一致,x的相加维度和y的相加维度的维度必须保持一致。比如,assi是[ dim1, ],y是[ dim1, dim2 ],x是[ XX, dim2]。

扫描二维码关注公众号,回复: 14615439 查看本文章

上述例子中,assi是[3,],y是[3,3],x是[3,3],满足维度要求。

那么该函数的实际意义就是:

按照assi的指定,将y的依次项加到x的指定index上去,本例子中assi是[1, 1, 2],就是说将y的第零个向量,第一个向量,第二个向量,分别加到x的index为1,1,2的向量上面去。

得到结果如下:

猜你喜欢

转载自blog.csdn.net/me_yundou/article/details/124252187
今日推荐