使用BN时ValueError: expected 2D or 3D input (got 4D input)的可能原因

版权声明：转载注明出处 https://blog.csdn.net/york1996/article/details/84200548

可能原因在于应该使用

BatchNorm2d

而你使用了

BatchNorm1d

---

如果是BatchNorm1d的话，input的形状应该是：

Input: :math:`(N, C)` or :math:`(N, C, L)`

---

如果是BatchNorm2d的话，input的形状应该是：

Input:  :math:`(N, C, H, W)`

---

除此之外，还有BatchNorm3d，input的形状应该是：

Input:  :math:`(N, C, D, H, W)`

---