package the function of nn.Embedding, nn.Dropout() in pytorch for use - View it on GitHub