Могу ли я использовать слой BatchNornalization после слоя conv3d?

Question

В keras.oi в разделе BatchNormalization, Аргументы: ось: целое число, ось, которая должна быть нормализована (обычно это ось объектов).Например, после слоя Conv2D с data_format = "channel_first" установите значение оси = 1 в BatchNormalization.

Означает ли это, что его можно использовать только после слоя Conv2D?Если в моей модели есть слои Conv3D, могу ли я по-прежнему использовать BatchNormalization (axis = 1)?

Answer 1

Если у вас есть conv3D, вы должны быть осторожны, чтобы выбрать правильную ось для нормализации.Как упомянуто в документации, если вы выбираете «сначала каналы» для вашего conv3d, то вы можете выполнить BatchNormalization с осью = 1 для трехмерного вектора (поскольку положение каналов будет таким же, как и для conv2D)