class_docstr: Wrapped nn.Embedding layer to allow for weight initialization.
class_docstr_tok: ["Wrapped","nn.Embedding","layer","to","allow","for","weight","initialization","."]
code: def forward(self, src):
      return super().forward(src) * self.ninp_sqrt
  
imports: import math
  import torch
  import torch.nn
  from fairscale.nn.moe.moe_layer import MOELayer
  from fairscale.nn.moe.top2gate import Top2Gate
label_desc: Use image node_id 2 for calling the EmbeddingLayer obj's underlying member method code with example usage: obj.forward(src) and returns: unknown