Keras批處理規范化-哪個軸？

Question

我的數據形狀如下：

(50000, 28, 28)

它是2D圖像，第一維是樣本數。 我的第一層確實變平。 我還想添加BatchNormalization層：

model.add(ll.InputLayer([28, 28]))

model.add(ll.Flatten())

# network body

model.add(ll.Dense(1000, kernel_regularizer=regularizers.l2(0.01), activation='elu'))
model.add(ll.BatchNormalization())

軸通過的正確值是多少？

Answer 1

通常，您想對批次上的特征尺寸進行規格化，因此，對於平面輸入，應在BatchNormalization中將axis=-1設置。