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

来源：3-5 实战自定义损失函数与DenseLayer回顾

慕田峪7306343

2019-09-27

写回答

1回答

正十七

2019-10-24

同学你好，关于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

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程

相似问题

可以把build函数定义weight和bias的代码放到__init__函数中吗？

回答 1

自定义损失函数如何再添加另一个参数呢？

回答 1

关于自定义模型的一个问题

回答 1

请问自定义损失函数的输入如果是多维的应该怎么写呢？

回答 2

feature_column自定义

回答 1

打开慕课网App查看更多内容