weixin_43496455的博客

torch.stack()跟torch.scatter()类似，scatter()请看我的上一篇博客：https://blog.youkuaiyun.com/weixin_43496455/article/details/103870889贴个torch.stack()官方文档的截图dim代表沿着哪个维度进行堆叠举个例子：dim=0时：（dim不写时，默认为0）a: 2x3 ; b: 2x3 ...

scatter()、scatter_()scatter() 和 scatter_() 的作用一样。不同之处在于 scatter() 不会直接修改原来的 Tensor，而 scatter_() 会在原来的基础上对Tensor进行修改。torch.scatter()官方文档scatter(dim, index, src)将src中数据根据index中的索引按照dim的方向进行填充。dim=0...

torch.cat()torch.cat是将两个张量（tensor）拼接在一起，cat是concatnate的意思，即拼接。其中dim代表维度，0代表行，1代表列。index_select()index_select(x, 1, indices)1代表dim(维度)为1，即列。indices是筛选的索引序号。当dim为0时，就是筛选行的索引，当dim为1时，就是筛选列的索引。注...