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

通过查看代码找到了问题所在是因为使用了不正确的BatchNorm函数原因分析我们可以看到官方文档对于BatchNorm1d的解释BatchNorm1d显然需要输入的是二维或者三维，但是我们有一个四维的输入那么我们可以查看BatchNorm2d的官方解释BatchNorm2d可以接受四维输入，解决方法根据上述分析，把BatchNorm1d改为BatchNorm2d即可...

yezhangyinge

16007人浏览 · 2021-05-01 17:12:37

yezhangyinge  ·  2021-05-01 17:12:37 发布

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

---

原因分析

我们可以看到官方文档对于BatchNorm1d的解释
BatchNorm1d

显然需要输入的是二维或者三维，但是我们有一个四维的输入
那么我们可以查看BatchNorm2d的官方解释
BatchNorm2d

可以接受四维输入，

---

解决方法

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

# python # 机器学习 # pytorch

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

加入社区

更多推荐

• · 软考中级-软件设计师 UML图详解（ 类图，对象图，用例图，序列图，通信图，状态图，活动图，构件图，部署图）
• · 机器学习 决策树-分类
• · 【SD教程】超详细AI绘画提示词语法讲解！

软考中级-软件设计师 UML图详解（ 类图，对象图，用例图，序列图，通信图，状态图，活动图，构件图，部署图）

2048 AI社区

机器学习 决策树-分类

2048 AI社区

【SD教程】超详细AI绘画提示词语法讲解！

2048 AI社区

欢迎加入社区

取消 确定