如何在tensorflow中实现图像的直方图均衡？

import tensorflow as tf
import numpy as np
import cv2
import matplotlib.pyplot as plt

image = cv2.imread('image.jpg')
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

histogram, bins = np.histogram(gray_image.flatten(), 256, [0, 256])

cdf = histogram.cumsum()
cdf_normalized = cdf * histogram.max() / cdf.max()

equalized_image = np.interp(gray_image.flatten(), bins[:-1], cdf_normalized)
equalized_image = equalized_image.reshape(gray_image.shape)

plt.subplot(121), plt.imshow(gray_image, cmap='gray'), plt.title('Original Image')
plt.subplot(122), plt.imshow(equalized_image, cmap='gray'), plt.title('Equalized Image')
plt.show()