code: def extract_embed(node_embed, input_nodes):
      emb = {}
      for ntype, nid in input_nodes.items():
          nid = input_nodes[ntype]
          emb[ntype] = node_embed[ntype][nid]
      return emb
  
file_path: dgl/examples/pytorch/rgcn-hetero/entity_classify_mb.py
imports: import argparse
  import itertools
  import time
  import dgl
  import numpy
  import torch
  import torch.nn.functional
  from dgl.data.rdf import AIFBDataset, AMDataset, BGSDataset, MUTAGDataset
  from model import EntityClassify, RelGraphEmbed
label_desc: Use image node_id 1 for calling a global function with example usage: extract_embed(node_embed, input_nodes) and returns: emb