code: def __init__(
      self,
      word_size,
      position_embeddings_size,
      word_type_size,
      hidden_size,
      dropout_prob,
  ):
      super(BertEmbeddings, self).__init__()
      self.word_embeddings = nn.Embedding(
          word_size, hidden_size, padding_idx=0
      )
      self.position_embeddings = nn.Embedding(
          position_embeddings_size, hidden_size
      )
      self.token_type_embeddings = nn.Embedding(
          word_type_size, hidden_size
      )
      self.layernorm = nn.LayerNorm(hidden_size, epsilon=1e-8)
      self.dropout = nn.Dropout(dropout_prob)
  
imports: import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddle.nn import ReLU, Swish, GELU
  import math
  from ppdet.core.workspace import register
  from ..shape_spec import ShapeSpec
label_desc: Use image node_id 1 to create a new BertEmbeddings object from inherited base classes: nn with example: obj = BertEmbeddings(word_size, position_embeddings_size, word_type_size, hidden_size, dropout_prob)