code: def kl_divergence(x, num_experts):
      p = (importance(x) + 1e-10) / x.shape[0]
      q = torch.full(
          size=(num_experts,),
          fill_value=1.0 / num_experts,
          device=p.device,
      )
      divergence = torch.sum(p * torch.log(p / q))
      return divergence
  
label_desc: Use image node_id 1 for calling a global function with example usage: kl_divergence(x, num_experts) and returns: divergence