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 __init__(self, ntoken, ninp, initrange):
      super().__init__(ntoken, ninp)
      self.ninp_sqrt = math.sqrt(ninp)
      self.weight.data.uniform_(-initrange, initrange)
  
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 1 to create a new EmbeddingLayer object from inherited base classes: nn with example: obj = EmbeddingLayer(ntoken, ninp, initrange)