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]
'''