在tensorflow中调整图像大小？

import tensorflow as tf

# 假设img是一个形状为[height, width, channels]的图像张量
# target_size是一个包含目标高度和宽度的列表，例如[256, 256]
def resize_image(img, target_size):
    resized_img = tf.image.resize(img, target_size, method=tf.image.ResizeMethod.BILINEAR)
    return resized_img

# 使用示例
# 假设image是一个形状为[height, width, channels]的图像张量
image = ...

# 调整图像大小为256x256
resized_image = resize_image(image, [256, 256])

# 打印调整后的图像形状
print(resized_image.shape)