tf.keras.layers.Resizing 示例 改变维度的层

tf.keras.layers.Resizing 示例 改变维度的层

import tensorflow as tf

# 创建一个输入的张量
inputs = tf.random.normal([ 
    200,  # 维度变化
    193,  # 维度变化
    1  # 维度不变
])
print("输入维度：", inputs.shape)

输入维度： (200, 193, 1)

resiz_layer = tf.keras.layers.Resizing(
    32,  # 要改变的第一个维度
    32  # 要改变的第二个维度
)
output = resiz_layer(inputs)
print("输出维度：", output.shape)

输出维度： (32, 32, 1)