PyTorch
PyTorch学习中的细节知识点。
野草.
这个作者很懒,什么都没留下…
展开
-
PyTorch中scatter()和scatter_()函数的作用
本文讲的是我对PyTorch中scatter()函数的理解。原创,转载请标明来源。一言以蔽之:修改tensor中的指定位置的值。函数scatter(dim, index, src)dim: 索引的维度。按照i, j, k, ...的哪个方向进行索引 index: 索引。可以是一个tensor,存储需要改的元素的位置的tensor src: 用src中的值来修改。可以是tensor;可以是一个数字,用同样的数字写入tensorscatter() 和 scatter_() 函数..原创 2020-10-04 15:20:44 · 1475 阅读 · 0 评论 -
torch.gather()的理解
函数定义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...原创 2020-03-18 03:08:17 · 450 阅读 · 1 评论 -
jupyter notebook 中方括号(中括号)在markdown中不能正常显示的处理
在jupyternotebook中,若在markdown模式下输入a[0][0] = 0往往不会显示正确的结果。这时可以使用 转义 (反斜杠)来解决这个问题:a\[0\]\[0\] = 0就可以正确显示了。另外,因为jupyter notebook是一个开源的Web应用程序,所以使用html转义字符也可以解决:( [ ] 的HTML转义字符分别为[和...原创 2020-03-18 02:32:59 · 3067 阅读 · 0 评论