自定义层时build函数中add_weight的initializer参数如何自定义？

Answer 1

正十七 2019-10-24 22:20:44

同学你好，关于initializer参数，如果你是想换这用其他的initializer，那么需要在自定义层的构造函数中添加对应的参数及成员变量。

如果你是想用自己定义的initializer，那么需要继承自tf.keras.initializers.Initializer类，具体可以参考VarianceScaling类：https://github.com/tensorflow/tensorflow/blob/r2.0/tensorflow/python/ops/init_ops_v2.py#L702-L720