szwlh-c

VIG的代码疑问

1

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

您好，论文中说‘we simply replace SAM with a Conv-LReLU block.’，X的形状是H * W * C，D-GCN需要的形状是C*D。只用卷积好像无法降低张量的维度，这里是怎么转换形状的呢？