Solved by ptrblck in post #3 
              
              
                You could feed the data as a single tensor to the nn.Embedding layer. 
However, I would use separate embeddings, as your input data would have completely different ranges. 
nn.Embedding layers accept inputs containing values in [0, nb_embeddings-1]. 
If your year data contains values in e.g. [1988,…