在tf.keras.Model (命令式应用编程接口)中获取输入形状

import tensorflow as tf

# 定义模型
model = tf.keras.models.Sequential([
    tf.keras.layers.Input(shape=(32, 32, 3)),  # 输入层，指定输入形状为(32, 32, 3)
    tf.keras.layers.Dense(64, activation='relu')  # 全连接层
])

# 打印输入形状
print(model.input_shape)

import tensorflow as tf

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.input_layer = tf.keras.layers.Input(shape=(32, 32, 3))
        self.dense_layer = tf.keras.layers.Dense(64, activation='relu')

    def call(self, inputs):
        x = self.input_layer(inputs)
        x = self.dense_layer(x)
        return x

# 创建模型实例
model = MyModel()

# 打印输入形状
print(model.input_shape)