code: def tree_map(f):
      """Transforms `f` into a tree-mapped version."""
  
      def mapped_f(*structures):
          return tree.map_structure(f, *structures)
  
      return mapped_f
  
file_path: acme/acme/utils/tree_utils.py
imports: from typing import Sequence, List, TypeVar, Any
  import numpy
  import tree
label_desc: Use image node_id 6 for calling a global function with example usage: tree_map(f) and returns: mapped_f, tree