keras 共享權重層

在寫CBAM時遇到問題,maxpooling和average pooling要共享一個全連接,但是不會寫這種共享一個層的。 寫法: 先定義出層不用加輸入,相當於一個函數 然後用的時候直接調用就好了。 感謝大神!!參考:https://blog.csdn.net/ngadminq/article/details/107005092
相關文章
相關標籤/搜索