class_docstr: Word Embedding module
class_docstr_tok: ["Word","Embedding","module"]
code: def __init__(self, vocab_size, dim_model):
      super(Embeddings, self).__init__()
      self.lut = nn.Embedding(vocab_size, dim_model)
      self.dim_model = dim_model
  
file_path: dgl/examples/pytorch/transformer/modules/embedding.py
imports: import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new Embeddings object from inherited base classes: nn with example: obj = Embeddings(vocab_size, dim_model)