tensorflow學習筆記--embedding_lookup()用法

版權宣告：本文為博主原創文章，未經博主允許不得轉載。 https://blog.csdn.net/u013041398/article/details/60955847

embedding_lookup( )的用法 
關於tensorflow中embedding_lookup( )的用法，在Udacity的word2vec會涉及到，本文將通俗的進行解釋。

首先看一段網上的簡單程式碼：

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14

程式碼中先使用palceholder定義了一個未知變數input_ids用於儲存索引，和一個已知變數embedding，是一個5*5的對角矩陣。 
執行結果為：

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12

簡單的講就是根據input_ids中的id，尋找embedding中的對應元素。比如，input_ids=[1,3,5]，則找出embedding中下標為1,3,5的向量組成一個矩陣返回。 

如果將input_ids改寫成下面的格式：

• 1
• 2

輸出結果就會變成如下的格式：

• 1
• 2
• 3
• 4
• 5
• 6

對比上下兩個結果不難發現，相當於在np.array中直接採用下標陣列獲取資料。需要注意的細節是返回的tensor的dtype和傳入的被查詢的tensor的dtype保持一致；和ids的dtype無關。