Are different labels possible for one datapoint?
To include your additional data you would have to encode it somehow. The simplest approach would be to use one-hot encoding which, however, is relatively sparse. Another approach would be to use nn.Embedding 2 as this will not be one-hot encoded but in a dense way.