TensorFlow.js tf.layers.getConfig() 方法

tf.layers.getConfig() 方法返回当前层的配置参数。这些配置参数可以用于编写自定义层或在恢复模型权重时使用。

语法

tf.layers.getConfig(): LayerConfig;

返回值：一个包含当前层配置参数的 LayerConfig 对象。

示例

const input = tf.input({ shape: [4] });
const denseLayer = tf.layers.dense({ units: 8, activation: 'relu' });
const output = denseLayer.apply(input);

console.log(denseLayer.getConfig());

输出：

{
  name: 'dense',
  trainable: true, 
  batchInputShape: [null, 4],
  dtype: 'float32',
  units: 8,
  kernelInitializer:
   { className: 'VarianceScaling',
     config: { scale: 1, mode: 'fanAvg', distribution: 'uniform' } },
  biasInitializer:
   { className: 'Zeros', config: {} },
  activation: 'relu',
  useBias: true,
  kernelRegularizer: null,
  biasRegularizer: null,
  activityRegularizer: null,
  kernelConstraint: null,
  biasConstraint: null
}

参考资料

• tf.layers.getConfig() (TensorFlow.js官方文档)