2015-11-23 45 views
2

我試圖實現一個lambda層,其中由氧層執行的功能正在執行transpose(P)*x,其中xM行數組和P每個條目是1/MKeras/Theano - 如何通過在載體LAMBDA層相乘

我遇到的問題是,我不知道如何創建一個靜態數組,每個條目P設置爲1/M。我知道我可以使用爲張量定義的shape()函數獲得M,但是我不確定如何創建矢量。

回答

1

我認爲你是過度複雜的事情。如果你想用一個填充了常量的矩陣乘,你可以簡單地除以一個標量,然後在你的輸入矩陣上廣播,例如,

Lambda(lambda x: x/M) 

M何處可以被定義爲

from keras import backend as K 
M = K.shape(x)[0] 

給予

Lambda(lambda x: x/K.shape(M)[0])