code: def message_func(edges):
      assert F.ndim(edges.src["h"]) == 2
      assert F.shape(edges.src["h"])[1] == D
      return {"m": edges.src["h"]}
  
imports: import warnings
  from collections import defaultdict
  import backend
  import dgl
  import networkx
  import numpy
  from utils import parametrize_idtype
label_desc: Use image node_id 1 for calling a global function with example usage: message_func(edges) and returns: dict