VIG的代码疑问

[szwlh-c]

你好，代码x = torch.cat([x.unsqueeze(2), x_j.unsqueeze(2)], dim=2).reshape(b, 2 * c, n, _)会对x和x''做一个拼接，然后做线性变换。但是论文中好像没有描述这一点。 请问这是为什么呢？是我哪里弄错了吗？

[iamhankai]

不好意思，论文里写错了，最新版已经更正：https://arxiv.org/pdf/2206.00272.pdf