TensorFlow Keras指南：自定义图层的get_config方法是否不更新图层的配置？

Question

在TensorFlow Keras指南的custom layers部分下，有get_config方法的可选实现：

  def get_config(self):
    base_config = super(MyLayer, self).get_config()
    base_config['output_dim'] = self.output_dim

使用一些伪数据运行完整代码，然后查看在model.get_config()之后调用model.fit(...产生的配置，将产生以下输出：

[{'class_name': 'MyLayer', 'config': None},
 {'class_name': 'Activation',
 'config': {'name': 'activation_38',
  'trainable': True,
  'dtype': 'float32',
  'activation': 'softmax'}}]

我的问题是：由于output_dim方法中的最终赋值，自定义层“ MyLayer”的配置是否不包含get_config字段？非常感谢。

Answer 1

没有get语句的return方法吗？

这就是None的原因。

def get_config(self):
    base_config = super(MyLayer, self).get_config()
    base_config['output_dim'] = self.output_dim
    return base_config