pytorch出现RuntimeError,维度不匹配

数据是N1121145121的numpy数组,N是样本数量,设置的batch为4
运行的时候出现了这个错误RuntimeError: Expected 5-dimensional input for 5-dimensional weight [64, 1, 7, 7, 7], but got 6-dimensional input of size [4, 2, 1, 121, 145, 121] instead
用torch.squeeze也没啥变化
不知道是哪个环节出现问题,期待解答,谢谢!

emm. 很明显的问题 维度不一致
要么改网络 要么改输入

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^