标签: tensorflow keras keras-layer
按照以下文档->
https://keras.io/layers/writing-your-own-keras-layers/
它用于初始化权重矩阵。但是我怀疑为什么不只在调用函数本身中进行这种初始化?
答案 0 :(得分:0)
Build函数仅用于更好地抽象和结构化代码。最后,从 call 调用构建函数,在此我们知道输入形状。如果您已经知道输入形状,也可以在init函数中初始化权重