Word2vec Pytorch Gpu