code: def u_mul_e_udf(edges):
      return {"m": edges.src["h"] * edges.data["w"]}
  
imports: import os
  import dgl
  import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
  import dgl.data
label_desc: Use image node_id 2 for calling a global function with example usage: u_mul_e_udf(edges) and returns: dict