Question about GraphBlock

[Tyler-1cloud]

当我对这个块实例化运行的时候，出现以下报错： return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (4x16 and 1x64)
以下图片是实例化参数的取值

当C_out = d_model 时，没有报错，但是在代码设置里C_out 表示的是输出变量的大小,在MS预测中为1 ，导致出现报错

[YoZhibo]

您好，感谢关注！抱歉最近比较忙碌。c_out=1时squeeze()会去除多个维度导致报错。您可以尝试将其改为squeeze(-1)，修改后应该能够正常运行。