PyTorch Embedding

The input of embedding layer is a vector of indices, the output of embedding layer is a matrix of corresponding embeddings.