code: def gather(consts: torch.Tensor, t: torch.Tensor):
      """Gather consts for $t$ and reshape to feature map shape"""
      c = consts.gather(-1, t)
      return c.reshape(-1, 1, 1, 1)
  
label_desc: Use image node_id 1 for calling a global function with example usage: gather(consts, t) and returns: c