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

通常，您希望对批次进行特征尺寸标准化，因此，对于平面输入，应在axis=-1中设置BatchNormalization。