class_docstr: Word Embedding module
class_docstr_tok: ["Word","Embedding","module"]
code: def forward(self, x):
      return self.lut(x) * np.sqrt(self.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 2 for calling the Embeddings obj's underlying member method code with example usage: obj.forward(x) and returns: unknown