code: def metadata_like(rays, metadata_id):
      """Create a metadata array like a ray batch."""
      return jnp.full_like(
          rays[..., :1], fill_value=metadata_id, dtype=jnp.uint32
      )
  
file_path: nerfies/nerfies/model_utils.py
imports: from flax import linen
  from flax import optim
  from flax import struct
  from jax import lax
  from jax import random
  import jax.numpy
label_desc: Use image node_id 10 for calling a global function with example usage: metadata_like(rays, metadata_id) and returns: jnp