tf.keras.Sequential(
    layers=None, name=None
)

# 5層網絡模型
model = tf.keras.Sequential([
    tf.keras.layers.Dense(256, activation=tf.nn.relu),
    tf.keras.layers.Dense(128, activation=tf.nn.relu),
    tf.keras.layers.Dense(64, activation=tf.nn.relu),
    tf.keras.layers.Dense(32, activation=tf.nn.relu),
    tf.keras.layers.Dense(10)
])

class My_Model(tf.keras.Model):  # 繼承Model

    def __init__(self):
        """
        初始化
        """
        
        super(My_Model, self).__init__()
        self.fc1 = My_Dense(784, 256)  # 第一層
        self.fc2 = My_Dense(256, 128)  # 第二層
        self.fc3 = My_Dense(128, 64)  # 第三層
        self.fc4 = My_Dense(64, 32)  # 第四層
        self.fc5 = My_Dense(32, 10)  # 第五層

    def call(self, inputs, training=None):
        """
        在Model被調用的時候執(zhí)行
        :param inputs: 輸入
        :param training: 默認為None
        :return: 返回輸出
        """
        
        x = self.fc1(inputs)
        x = tf.nn.relu(x)
        x = self.fc2(x)
        x = tf.nn.relu(x)
        x = self.fc3(x)
        x = tf.nn.relu(x)
        x = self.fc4(x)
        x = tf.nn.relu(x)
        x = self.fc5(x)

        return x

class My_Dense(tf.keras.layers.Layer):  # 繼承Layer

    def __init__(self, input_dim, output_dim):
        """
        初始化
        :param input_dim:
        :param output_dim:
        """

        super(My_Dense, self).__init__()

        # 添加變量
        self.kernel = self.add_variable("w", [input_dim, output_dim])  # 權重
        self.bias = self.add_variable("b", [output_dim])  # 偏置

    def call(self, inputs, training=None):
        """
        在Layer被調用的時候執(zhí)行, 計算結果
        :param inputs: 輸入
        :param training: 默認為None
        :return: 返回計算結果
        """

        # y = w * x + b
        out = inputs @ self.kernel + self.bias

        return out

import tensorflow as tf

def pre_process(x, y):

    # 轉換x
    x = 2 * tf.cast(x, dtype=tf.float32) / 255 - 1  # 轉換為-1~1的形式
    x = tf.reshape(x, [-1, 32 * 32 * 3])  # 把x鋪平

    # 轉換y
    y = tf.convert_to_tensor(y)  # 轉換為0~1的形式
    y = tf.one_hot(y, depth=10)  # 轉成one_hot編碼

    # 返回x, y
    return x, y

def get_data():
    """
    獲取數據
    :return:
    """

    # 獲取數據
    (X_train, y_train), (X_test, y_test) = tf.keras.datasets.cifar10.load_data()

    # 調試輸出維度
    print(X_train.shape)  # (50000, 32, 32, 3)
    print(y_train.shape)  # (50000, 1)

    # squeeze
    y_train = tf.squeeze(y_train)  # (50000, 1) => (50000,)
    y_test = tf.squeeze(y_test)  # (10000, 1) => (10000,)

    # 分割訓練集
    train_db = tf.data.Dataset.from_tensor_slices((X_train, y_train)).shuffle(10000, seed=0)
    train_db = train_db.batch(batch_size).map(pre_process).repeat(iteration_num)  # 迭代20次

    # 分割測試集
    test_db = tf.data.Dataset.from_tensor_slices((X_test, y_test)).shuffle(10000, seed=0)
    test_db = test_db.batch(batch_size).map(pre_process)

    return train_db, test_db

class My_Dense(tf.keras.layers.Layer):  # 繼承Layer

    def __init__(self, input_dim, output_dim):
        """
        初始化
        :param input_dim:
        :param output_dim:
        """

        super(My_Dense, self).__init__()

        # 添加變量
        self.kernel = self.add_weight("w", [input_dim, output_dim])  # 權重
        self.bias = self.add_weight("b", [output_dim])  # 偏置

    def call(self, inputs, training=None):
        """
        在Layer被調用的時候執(zhí)行, 計算結果
        :param inputs: 輸入
        :param training: 默認為None
        :return: 返回計算結果
        """

        # y = w * x + b
        out = inputs @ self.kernel + self.bias

        return out


class My_Model(tf.keras.Model):  # 繼承Model

    def __init__(self):
        """
        初始化
        """

        super(My_Model, self).__init__()
        self.fc1 = My_Dense(32 * 32 * 3, 256)  # 第一層
        self.fc2 = My_Dense(256, 128)  # 第二層
        self.fc3 = My_Dense(128, 64)  # 第三層
        self.fc4 = My_Dense(64, 32)  # 第四層
        self.fc5 = My_Dense(32, 10)  # 第五層

    def call(self, inputs, training=None):
        """
        在Model被調用的時候執(zhí)行
        :param inputs: 輸入
        :param training: 默認為None
        :return: 返回輸出
        """

        x = self.fc1(inputs)
        x = tf.nn.relu(x)
        x = self.fc2(x)
        x = tf.nn.relu(x)
        x = self.fc3(x)
        x = tf.nn.relu(x)
        x = self.fc4(x)
        x = tf.nn.relu(x)
        x = self.fc5(x)

        return x

# 定義超參數
batch_size = 256  # 一次訓練的樣本數目
learning_rate = 0.001  # 學習率
iteration_num = 20  # 迭代次數
optimizer = tf.keras.optimizers.Adam(learning_rate=learning_rate)  # 優(yōu)化器
loss = tf.losses.CategoricalCrossentropy(from_logits=True)  # 損失
network = My_Model()  # 實例化網絡

# 調試輸出summary
network.build(input_shape=[None, 32 * 32 * 3])
print(network.summary())

# 組合
network.compile(optimizer=optimizer,
                loss=loss,
                metrics=["accuracy"])

if __name__ == "__main__":
    # 獲取分割的數據集
    train_db, test_db = get_data()

    # 擬合
    network.fit(train_db, epochs=5, validation_data=test_db, validation_freq=1)