在conv3d层之后可以使用BatchNornalization层吗?

时间:2019-06-22 00:41:41

标签: python keras-layer

在keras.oi的BatchNormalization下, 参数: axis:整数,应归一化的轴(通常是要素轴)。例如,在使用data_format =“ channels_first”的Conv2D图层之后,在BatchNormalization中设置axis = 1。

这是否意味着它只能在Conv2D层之后使用?如果我的模型具有Conv3D图层,是否仍可以使用BatchNormalization(axis = 1)?

1 个答案:

答案 0 :(得分:0)

如果有conv3D,则必须小心选择正确的轴进行标准化。如文档中所述,如果为conv3d选择“通道优先”,则可以在3D向量上使用axis = 1进行BatchNormalization(因为通道的位置与conv2D相同)