code: def dict_multimap(fn, dicts):
      first = dicts[0]
      new_dict = {}
      for k, v in first.items():
          all_v = [d[k] for d in dicts]
          if type(v) is dict:
              new_dict[k] = dict_multimap(fn, all_v)
          else:
              new_dict[k] = fn(all_v)
  
      return new_dict
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/src/transformers/models/esm/openfold_utils/tensor_utils.py
imports: from functools import partial
  from typing import List
  import torch
  import torch.nn
label_desc: Use image node_id 6 for calling a global function with example usage: dict_multimap(fn, dicts) and returns: new_dict