Imports a BatchNormalization layer from Keras.
KerasBatchNormalization
- param kerasVersion major keras version
- throws UnsupportedKerasConfigurationException Unsupported Keras config
getBatchNormalizationLayer
Constructor from parsed Keras layer configuration dictionary.
- param layerConfig dictionary containing Keras layer configuration
- throws UnsupportedKerasConfigurationException Unsupported Keras config
getOutputType
- param inputType Array of InputTypes
- return output type as InputType
- throws InvalidKerasConfigurationException Invalid Keras config
getNumParams
Returns number of trainable parameters in layer.
setWeights
- param weights Map from parameter name to INDArray.