2
我試圖實現一個lambda層,其中由氧層執行的功能正在執行transpose(P)*x
,其中x
與M
行數組和P
每個條目是1/M
。Keras/Theano - 如何通過在載體LAMBDA層相乘
我遇到的問題是,我不知道如何創建一個靜態數組,每個條目P
設置爲1/M
。我知道我可以使用爲張量定義的shape()
函數獲得M
,但是我不確定如何創建矢量。
我試圖實現一個lambda層,其中由氧層執行的功能正在執行transpose(P)*x
,其中x
與M
行數組和P
每個條目是1/M
。Keras/Theano - 如何通過在載體LAMBDA層相乘
我遇到的問題是,我不知道如何創建一個靜態數組,每個條目P
設置爲1/M
。我知道我可以使用爲張量定義的shape()
函數獲得M
,但是我不確定如何創建矢量。
我認爲你是過度複雜的事情。如果你想用一個填充了常量的矩陣乘,你可以簡單地除以一個標量,然後在你的輸入矩陣上廣播,例如,
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])