Interesting.  Thanks for the insight.
So if I want to use this for multiple embeddings together I’d need to write a custom loss function that handles this then?  Makes sense.  I’ll take a look at the implementation of cross_entropy.