scatter()、scatter_()
scatter() 和 scatter_() 的作用一样。
不同之处在于 scatter() 不会直接修改原来的 Tensor,而 scatter_() 会在原来的基础上对Tensor进行修改。
torch.scatter()官方文档
scatter(dim, index, src)将src中数据根据index中的索引按照dim的方向进行填充。
dim=0
————————————————
版权声明:本文为CSDN博主「又笨又懒的猪」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_43496455/article/details/103870889