keras 编写自己的Layer

1. Layer重写的基本思路? Keras层的骨架。只需要实现三个方法即可: build(input_shape): 这是你定义权重的地方。这个方法必须设self.built = True,可以通过调用super([Layer], self).build()完成。build,这个函数用来确立这个层都有哪些参数,哪些参数是可训练的哪些参数是不可训练的。 call(x): 这里是编写层的功能逻辑的地
相关文章
相关标签/搜索