embeddings =tf.nn.embedding_lookup(feature_embeddings, feat_index) # None * F * K

• 输入1：张量feature_embeddings维度为F*K
• 输入2：张量feat_index维度为None * F
• 输出： tf.nn.embedding_lookup()的维度为None * F * K

参考

ChenglongChen/tensorflow-DeepFM