pytorch文档介绍

scatter_.png
scatter_1.png

实验

在指标张量中指定的指标处将张量src的所有值写入self。对于src中的每个值,其输出索引由其在src中的index For dimension != dim和index For dimension = dim中对应的值指定。
src
scatter
函数作用:将src通过index填充到自身
如果dim=0,那么index中的值代表的就是要填充到的行数,index中的值所在的列代表的是要填充到的列数,index中该值所在的行数和列数,对应src中应该填充的值的行数和列数。
dim=0
如果dim=1,表示按行填充,index中存储的就是列号,index中值所在的行号和自身值作为列号,将对应index位置的src数值填充到对应的位置去。
其中,当dim=0时,index的列数要和待填充的矩阵列数相同;当dim=1时,index的行数要和待填充的矩阵行数相同。

最后修改:2021 年 10 月 02 日
如果觉得我的文章对你有用,请随意赞赏