Keras-如何为keras嵌入层实现权重总和

时间:2019-02-24 07:33:13

标签: keras embedding word-embedding keras-2

我正在使用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

(注意:重量可以训练或固定)

0 个答案:

没有答案