code: def read_image(filepath):
      """Returns an image tensor."""
      im_bytes = tf.io.read_file(filepath)
      im = tf.image.decode_image(im_bytes, channels=CHANNELS)
      im = tf.image.convert_image_dtype(im, tf.float32)
      return im
  
file_path: probability/tensorflow_probability/examples/sprites_dataset.py
imports: from collections import namedtuple
  import os
  import random
  import zipfile
  from absl import flags
  from six.moves import urllib
  import tensorflow.compat.v1
  from tensorflow.python.ops import lookup_ops
label_desc: Use image node_id 1 for calling a global function with example usage: read_image(filepath) and returns: im