iT邦幫忙

0

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

各位邦友好:
我想在Convolution layer之前對tensor的每一個elements做取二進位16bits的運算,function我已經寫好了,但我把function放到model裡面會有叫我feed a value for placeholder tensor的error。
https://ithelp.ithome.com.tw/upload/images/20200420/201264660Hzh9IHFTq.png
我的猜想是:在keras中是先宣告model再給model訓練資料,因為我的function沒有轉成layer,所以在宣告model的時候他就會跟我要input,我有試過用backend.lambda想將function轉成layer但會有相同的error。
希望有遇過相同問題的邦友們可以給我一些指點,謝謝!

尚未有邦友回答

立即登入回答