在Keras中使用2個張量輸入定義自定義層

Question

我想實現一個自定義層。 我的自定義層的2個輸入是2個張量，它們來自2個單獨的2D卷積層，有一個例子嗎？

Answer 1

由於您不需要訓練它，因此lambda函數也可以。 或者，您可以保留自定義圖層，並將其設置為False。 權重將永遠不會針對該層進行更新，並且您在此處所做的任何操作都將向前傳播到模型中的下一層，並且如注釋中所述，反向傳播將通過權重影響其他層。 因此，您的模型肯定會學到一些東西。

我個人建議使用自定義層，如果您稍后決定向該層添加一些知識並檢查結果。 您不能在Lambda函數中執行此操作。 如果添加一個（內核），則必須在“調用”方法中使用。 否則，您的模型將在訓練過程中引發錯誤。