PyTorch中的BatchNorm2d層

先來看看pytorch中對於類的定義：python CLASS torch.nn.BatchNorm2d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) 解析一下每一個參數的做用：學習 num_features：BatchNorm是針對每個通道作的，因此這裏應該填寫（N, C, H, W）中

>>阅读原文<<