給定一組輸入句子{
同時有一個給定的查詢句子q,用上面同樣的方法得出他的向量表示爲u。接下來計算q和輸入句子中每個子句的匹配程度即概率,公式如下:
其中p是一個行向量,u是一個列向量,m是輸入句子轉換後得到的矩陣,每一列代表一個句子。p中的每一個元素代表匹配的程度。
另外每個輸入句子有一個輸出向量
通過上面的式子得出一個回覆向量o
o是一個列向量,其實是矩陣c中的列向量加權和,權向量即爲p
最後根據恢復向量來生成最終的預測。在單層的網絡裏將回復向量和u相加,傳入一個最終的權重矩陣爲W(V乘以d)的softmax層,得出最後的輸出列向量a,a的長度爲V。所以最終的輸出結果是一個詞。