code: def _equalize_image(image: TensorLike, bins: int = 256) -> tf.Tensor:
      """Implements Equalize function from PIL using TF ops."""
      image = tf.stack(
          [
              _scale_channel(image, c, bins)
              for c in range(image.shape[-1])
          ],
          -1,
      )
      return image
  
file_path: addons/tensorflow_addons/image/color_ops.py
imports: import tensorflow
  from tensorflow_addons.utils.types import TensorLike, Number
  from tensorflow_addons.image.utils import to_4D_image, from_4D_image
  from tensorflow_addons.image.compose_ops import blend
  from typing import Optional
  from functools import partial
label_desc: Use image node_id 2 for calling a global function with example usage: _equalize_image(image, bins) and returns: image