code: def group_dict_by_key(cond, d):
      return_val = [dict(), dict()]
      for key in d.keys():
          match = bool(cond(key))
          ind = int(not match)
          return_val[ind][key] = d[key]
      return (*return_val,)
  
imports: import torch
  from torch import nn, einsum
  import torch.nn.functional
  from einops import rearrange, repeat
  from einops.layers.torch import Rearrange
label_desc: Use image node_id 1 for calling a global function with example usage: group_dict_by_key(cond, d) and returns: unknown