Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Libnd4j: embedding_lookup op needs to support multiple input arrays #6311
Currently, the embedding_lookup op assumes that all data is in a single array.
For TF import, we need this op (or a new op, if that's easier) to support multiple input arrays - i.e., a "split" input array (still with single indices array), with partitioning strategies "mod" and "div":