TensorFlow2.0學習筆記-4.模型的自定義

4.模型的自定義 4.1.自定義層 使用的主要數據結構是Layer 實現自定義層的最佳方法是擴展tf.keras.layers.Layer類並實現： • __init__ ：可以在其中進行所有與輸入無關的初始化，定義相關的層 • build： 知道輸入張量的形狀並可以進行其餘的初始化 • call： 在這裏進行前向傳播 注意：不一定需要在build中創建變量時，也可以在__init__中創建它們。

>>阅读原文<<