code: def fast_map_structure(func, *structure):
      """Faster map_structure implementation which skips some error checking."""
      flat_structure = (tree.flatten(s) for s in structure)
      entries = zip(*flat_structure)
      # Arbitrarily choose one of the structures of the original sequence (the last)
      # to match the structure for the flattened sequence.
      return tree.unflatten_as(
          structure[-1], [func(*x) for x in entries]
      )
  
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 1 for calling a global function with example usage: fast_map_structure(func) and returns: tree