如何将图像的张量保存为PNG？

import tensorflow as tf
import matplotlib.pyplot as plt

image_path = 'path_to_image.png'
image = tf.io.read_file(image_path)
image = tf.image.decode_image(image)

output_path = 'path_to_save_image.png'
tf.io.write_file(output_path, tf.image.encode_png(image))

import tensorflow as tf
import matplotlib.pyplot as plt

# 加载图像数据并将其转换为张量
image_path = 'path_to_image.png'
image = tf.io.read_file(image_path)
image = tf.image.decode_image(image)

# 将图像张量保存为PNG文件
output_path = 'path_to_save_image.png'
tf.io.write_file(output_path, tf.image.encode_png(image))