code: def _track_provenance_pytree(x, provenance: frozenset):
      return tree_map(
          partial(track_provenance, provenance=provenance), x
      )
  
file_path: pyro/pyro/ops/provenance.py
imports: from functools import partial, singledispatch
  from typing import Tuple
  import torch
  from torch.utils._pytree import tree_flatten, tree_map, tree_unflatten
label_desc: Use image node_id 3 for calling a global function with example usage: _track_provenance_pytree(x, provenance) and returns: tree_map