code: def identity(features: Dict[str, Any]) -> Dict[str, Any]:
      """tf.identity for nested dictionary of Tensors."""
      out = {}
      for k, v in features.items():
          if isinstance(v, tf.Tensor):
              out[k] = tf.identity(v)
          elif isinstance(v, dict):
              out[k] = identity(v)
          else:
              raise NotImplementedError(
                  f"{v}'s type that is unsupported by identity."
              )
  
      return out
  
imports: from typing import Any, Dict
  import tensorflow
label_desc: Use image node_id 4 for calling a global function with example usage: identity(features) and returns: out