为什么运行代码时在GatedCNNBlock中g,i,c = torch,split(self.fc1(x),self.split_indices, dim= 1)会出现报错runtimeError: mat1 and mat2 shapes cannot be multiplied (12800×28 and 64×340)，哪位科迷帮忙解决一下

[Muzixingxi]

为什么运行代码时在GatedCNNBlock中g,i,c = torch,split(self.fc1(x),self.split_indices, dim= 1)会出现报错runtimeError: mat1 and mat2 shapes cannot be multiplied (12800×28 and 64×340)，哪位科迷帮忙解决一下

[yuweihao]

Can I know which model? I will debug it.

[Muzixingxi]

我是将MambaOut中的整个GatedCNNBlock嵌入到其他模块出现该问题

[Muzixingxi]

这是否和模型输入有关？

[yuweihao]

I guess yes. Please carefully check whether the input format is [B, H, W, C], if not, please permute it.