标签: python tensorflow keras customization
我想更好地了解如何制作自定义的keras图层,但是我似乎无法在网上找到以下内容...。我正在寻找如何使用自定义的基础(前馈)keras图层进行复制层结构。
我在哪里可以找到它,或者该怎么做?
我问,因为我遇到的所有在线资源仅以其他较简单的层为例。
答案 0 :(得分:2)
您需要子类tf.keras.layers.Layer。以here为例,给出了SimpleDense层。
tf.keras.layers.Layer
SimpleDense