code: def __init__(self, weight_shape, dtype=torch.half):
      super(EmbeddingLayer, self).__init__()
      self.weight = Parameter(
          torch.empty(
              weight_shape[0],
              weight_shape[1],
              dtype=dtype,
              device=torch.cuda.current_device(),
          )
      )
  
imports: import torch
  from deepspeed import comm
  from torch import nn
  from torch.nn import functional
  from torch.nn.parameter import Parameter
label_desc: Use image node_id 1 to create a new EmbeddingLayer object from inherited base classes: nn with example: obj = EmbeddingLayer(weight_shape, dtype)