Pytorch Label Embedding