code: def forward(self, g):
      g.ndata["h"] = self.embedding(g.ndata["Z"])
      g.apply_edges(self.edge_init)
      return g
  
file_path: dgl/examples/pytorch/dimenet/modules/embedding_block.py
imports: import numpy
  import torch
  import torch.nn
  from modules.envelope import Envelope
  from modules.initializers import GlorotOrthogonal
label_desc: Use image node_id 4 for calling the EmbeddingBlock obj's underlying member method code with example usage: obj.forward(g) and returns: g