在keras.layers

时间:2019-02-21 01:04:35

标签: tensorflow keras keras-layer

按照以下文档->

https://keras.io/layers/writing-your-own-keras-layers/

它用于初始化权重矩阵。但是我怀疑为什么不只在调用函数本身中进行这种初始化?

1 个答案:

答案 0 :(得分:0)

Build函数仅用于更好地抽象和结构化代码。最后,从 call 调用构建函数,在此我们知道输入形状。如果您已经知道输入形状,也可以在init函数中初始化权重