qq_38333041的博客

本文讲的是我对PyTorch中scatter()函数的理解。 原创，转载请标明来源。 一言以蔽之：修改tensor中的指定位置的值。 函数 scatter(dim, index, src) dim: 索引的维度。按照i, j, k, ...的哪个方向进行索引 index: 索引。可以是一个tensor，存储需要改的元素的位置的tensor src: 用src中的值来修改。可以是tensor；可以是一个数字，用同样的数字写入tensor scatter() 和 scatter_() 函数..

函数定义 torch.gather(input, dim, index, out=None) 其中， input(tensor): 待操作数，源张量 dim(int): 维度。 index(LongTensor): 索引下标 out: 输出的张量 举例 dim=0 输入： a = torch.arange(0,16).view(4,4) print(a) index...

在jupyternotebook中，若在markdown模式下输入 a[0][0] = 0 往往不会显示正确的结果。这时可以使用 转义 （反斜杠）来解决这个问题： a\[0\]\[0\] = 0 就可以正确显示了。 另外，因为jupyter notebook是一个开源的Web应用程序，所以使用html转义字符也可以解决：（ [ ] 的HTML转义字符分别为[和...