如何在keras的model中加入自定義function(layer)

keras python custom layer keras.backend.lambda

white850611 2020-04-20 23:42:39 ‧ 1156 瀏覽

分享至

各位邦友好：
我想在Convolution layer之前對tensor的每一個elements做取二進位16bits的運算，function我已經寫好了，但我把function放到model裡面會有叫我feed a value for placeholder tensor的error。

我的猜想是：在keras中是先宣告model再給model訓練資料，因為我的function沒有轉成layer，所以在宣告model的時候他就會跟我要input，我有試過用backend.lambda想將function轉成layer但會有相同的error。
希望有遇過相同問題的邦友們可以給我一些指點，謝謝！