当前位置: 代码迷 >> 综合 >> ValueError: expected 2D or 3D input (got 4D input)
  详细解决方案

ValueError: expected 2D or 3D input (got 4D input)

热度:38   发布时间:2023-11-27 09:49:50.0

通过查看代码找到了问题所在是因为使用了不正确的BatchNorm函数


原因分析

我们可以看到官方文档对于BatchNorm1d的解释
BatchNorm1d
在这里插入图片描述
显然需要输入的是二维或者三维,但是我们有一个四维的输入
那么我们可以查看BatchNorm2d的官方解释
BatchNorm2d
在这里插入图片描述
可以接受四维输入,


解决方法

根据上述分析,把BatchNorm1d改为BatchNorm2d即可

  相关解决方案