1. 程式人生 > >使用BN時ValueError: expected 2D or 3D input (got 4D input)的可能原因

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

可能原因在於應該使用

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)`