在TensorFlow中,可以使用model.summary()
方法來打印模型的結構。下面是一個示例:
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
# 創建一個簡單的神經網絡模型
model = Sequential([
Dense(64, activation='relu', input_shape=(784,)),
Dense(10, activation='softmax')
])
# 打印模型的結構
model.summary()
運行上面的代碼,就可以打印出神經網絡模型的結構信息,包括每一層的名稱、類型、輸出形狀和參數數量等。