code: def _tf_broadcast_metadata_fn(item):
      """Broadcasts metadata to the ray shape."""
      shape = tf.shape(item["rgb"])
      item["metadata"] = jax.tree_map(
          lambda x: tf.broadcast_to(x, (shape[0], shape[1], 1)),
          item["metadata"],
      )
      return item
  
imports: import abc
  import functools
  import itertools
  from typing import Any, Iterable, Optional, Sequence, Union
  from absl import logging
  from flax import jax_utils
  import jax
  import numpy
  import tensorflow
  from nerfies import camera
  from nerfies import gpath
  from nerfies import image_utils
  from nerfies import tf_camera
  from nerfies import utils
  from tensorflow.python.data.util import nest
label_desc: Use image node_id 7 for calling a global function with example usage: _tf_broadcast_metadata_fn(item) and returns: item