PyTorch学习笔记之初识word_embedding

 import torch
 import torch.nn as nn
 from torch.autograd import Variable

 word2id = {'hello': 0, 'world': 1}
 # you have 2 words, and then need 5 dim each word
 embeds = nn.Embedding(2, 5)
 # we need variable, because we need use element of nn.Embedding
 hello_idx = torch.LongTensor([word2id['hello']])
 print(hello_idx)
 hello_idx = Variable(hello_idx)
 print(hello_idx)
 # achieve the initial word_embedding of 'hello'
 hello_embed = embeds(hello_idx)
 print(hello_embed)
 '''
  0
 [torch.LongTensor of size 1]

 Variable containing:
  0
 [torch.LongTensor of size 1]

 Variable containing:
  1.1842  0.6819 -0.8768 -1.5130  0.7650
 [torch.FloatTensor of size 1x5]
 '''