code: def normalize(image, dtype=tf.float32):
      """Normalizes the value of pixels in the given image.
  
      Args:
        image: `Tensor` representing an image binary of arbitrary size.
        dtype: Tensorflow data type, Data type of the image.
  
      Returns:
        A normalized image `Tensor`.
      """
      image = tf.cast(image, dtype=dtype)
      if dtype not in [tf.int32, tf.int64, tf.uint32, tf.uint64]:
          image /= tf.constant(255.0, shape=[1, 1, 1], dtype=dtype)
      return image
  
imports: import collections
  import dataclasses
  import functools
  import itertools
  from typing import Any, Callable, Dict, Iterator, Optional, Sequence, Union
  from absl import logging
  from flax.training import common_utils
  import jax
  import jax.numpy
  import numpy
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 8 for calling a global function with example usage: normalize(image, dtype) and returns: image