code: def forward(self, srcs, dsts):
      # srcs: sku_id, brand_id, shop_id, cate_id
      srcs = self.query_node_embed(srcs)
      dsts = self.query_node_embed(dsts)
  
      return srcs, dsts
  
imports: import torch
label_desc: Use image node_id 2 for calling the EGES obj's underlying member method code with example usage: obj.forward(srcs, dsts) and returns: srcs,  dsts