@Alessandro Suglia I think this feature is useful, unfortunately, it is not supported now. One solution to get the same result, but slower, is to search twice. as below
lookup_result = tf.nn.embedding_lookup(emb, index) masked_emb = tf.concat(0, [tf.zeros([1, 1]), tf.ones([emb.get_shape()[0] - 1, 1]) mask_lookup_result = tf.nn.embedding_lookup(masked_emb, index) lookup_result = tf.mul(lookup_result, mask_lookup_result)
allen source share