Keras 自定义层

简单记录如下keras的自定义层的设置: 主要是三个方法的定义,以我本身的代码为例吧! 首先定义一个类,这个类继承了Keras.engine.topology.Layer。 1.首先是一些必要参数的初始化,这些参数的初始化写在def __init__(self,)中,而后是一些参数的初始化,记得最后要继承Layer中的一些初始化参数。 2.这部分主要是编写一些要更新的参数def build(sel
相关文章
相关标签/搜索