Pytorch Embedding Example