Pytorch word2vec. I am new to Pytorch and wanted your help.