我正在使用keras嵌入层。我想为嵌入设置权重总和(可训练或固定),但是我不确定该怎么做。
例如:
我的输入是文档表示[wID1, wID2, wID3]
的单词ID:
Input_vector = [[5, 1, 1]]
嵌入层[e1, e2, e3]
的输出:
Embedding layer = [[0.708, 0.129, 0.101],
[0.108, 0.459, 0.007],
[0.108, 0.459, 0.007]]
如果weight = [w1, w2, w3]
,我想按以下方式实现它:
Weight_sum = w1 * e1 + w2 * e2 + w3 * e3
(注意:重量可以训练或固定)