更新tensor指定位置的值 pytorch scatter_add_和scatter_
更新tensor指定位置的值可以使用tensor.scatter_add_(dim,index,src)官网上给出了一个例子:>>> x = torch.rand(2, 5)>>> xtensor([[0.7404, 0.0427, 0.6480, 0.3806, 0.8328], [0.7953, 0.2009, 0.9154, 0.6782, 0.9620]])>>> torch.ones(3, 5).scat...
原创
2020-06-22 11:30:54 ·
7442 阅读 ·
1 评论