tensorflow模型參數(shù)如何查看

小億
114
2024-04-01 14:19:03

在TensorFlow中，可以通過(guò)調(diào)用model.summary()方法來(lái)查看模型的參數(shù)。示例如下：

import tensorflow as tf

model = tf.keras.Sequential([
    tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)),
    tf.keras.layers.Dense(10, activation='softmax')
])

model.summary()

執(zhí)行以上代碼將會(huì)輸出模型的參數(shù)信息，包括每一層的名稱、類型、輸出shape以及可訓(xùn)練參數(shù)的數(shù)量。

0 踩