code: def apply_edges(self, edges):
      h_e = edges.data["h"]
      h_u = edges.src["h"]
      h_v = edges.dst["h"]
      score = self.W(th.cat([h_e, h_u, h_v], -1))
      return {"score": score}
  
file_path: dgl/examples/pytorch/gas/model.py
imports: import dgl.function
  import torch
  import torch.nn
  from dgl.nn.functional import edge_softmax
label_desc: Use image node_id 2 for calling the MLP obj's underlying member method code with example usage: obj.apply_edges(edges) and returns: dict