텐서플로우(Tensorflow) tf.nn.embedding_lookup() 이란?

        

import tensorflow as tf

#tf.nn.embedding_lookup 이란 첫번째 전달 인자에서 두번째 전달 인자의 index를 lookup(관찰, 추출)하는 것.

# [batch_size, dim]

self.user_embeddings = tf.nn.embedding_lookup(self.user_emb_matrix, self.user_indices)

tf.nn.embedding_lookup 이란 리스트의 크기가 클 경우, 원하는 index를 탐색하고 추출하기 위해 사용되는 함수이다.

전달인자로는 리스트, 탐색할 index리스트 이다. 

 

따라서 self.user_emb_matrix라는 큰 크기의 리스트에서 self.user_indeices의 index들을 탐색하여 반환한다.

 

조금 더 쉬운 예시