DL4J中文文档/Keras模型导入/归一化层

 

KerasBatchNormalization（Keras批量归一化层）

[源码]

从Keras导入一个BatchNormalization层。

KerasBatchNormalization

public KerasBatchNormalization(Integer kerasVersion) throws UnsupportedKerasConfigurationException 

从KerasLayer传递构造函数

• 参数 kerasVersion Keras主版本
• 抛出 UnsupportedKerasConfigurationException 不支持的 Keras 配置

getBatchNormalizationLayer

public BatchNormalization getBatchNormalizationLayer() 

获取BatchNormalization层

 

public KerasBatchNormalization(Map layerConfig)
            throws InvalidKerasConfigurationException, UnsupportedKerasConfigurationException

来自解析的keras层配置字典的构造函数。

• 参数 layerConfig 包含keras层配置的字典
• 抛出 InvalidKerasConfigurationException 无效的 Keras 配置
• 抛出 UnsupportedKerasConfigurationException 不支持的无效的 Keras 配置

getOutputType

public InputType getOutputType(InputType... inputType) throws InvalidKerasConfigurationException 

获取层输出类型

• 参数 inputType 输入类型数组
• 返回 按输入类型返回输出类型
• 抛出 InvalidKerasConfigurationException 无效的 Keras 配置

getNumParams

public int getNumParams() 

返回层中可训练参数的数目。

 

setWeights

public void setWeights(Map weights) throws InvalidKerasConfigurationException 

设置层的权重

• 参数 weights 从参数名映射到INDArray。