java icon indicating copy to clipboard operation
java copied to clipboard

Embeddings Lookup

Open danilojsl opened this issue 4 years ago • 3 comments

I am working with tensorflow-java version 0.2.0 trying to find some feature similar to tf.nn.embedding_lookup.

I found operations LookupTableInsert and LookupTableFind. Are any of these similar to tf.nn.embedding_lookup? In addition, these operations require a tableHandle argument which I don't know how to fill in.

Could you please guide me on how to use these operations?

Thanks

danilojsl avatar May 04 '21 13:05 danilojsl

tf.nn.embedding_lookup is a layer of python over the top of a gather call, which we have in the core ops. There's a bit of complexity there as the thing it looks up in could be partitioned and I've not traced that through yet. We don't have a wrapper around it at the moment, if you want to work on one that would be great.

Craigacp avatar May 04 '21 13:05 Craigacp

Hey @danilojsl , I know this issue is pretty old. But is there any appropriate feature similar to tf.nn.embedding_lookup in Java bindings yet? If so, could you let me know the function/operation name?

AgnidiptoSinha avatar Sep 22 '24 08:09 AgnidiptoSinha

We have not added a wrapper around the gather ops yet.

Craigacp avatar Sep 22 '24 15:09 Craigacp