Nn.embedding with multiple outputs?

I asked this in another thread but realise it probably warrants it’s own.

Already answered in the other thread:

It was way simpler than I expected.