Dense參數的units和input_dim意思?
2019/07/15 下午 06:26
機器學習共學討論版
施渝陽
觀看數:470
回答數:1
收藏數:0
ml100-2
ml100-2-d70
model.add(Dense(units=256,
input_dim=784,
kernel_initializer='normal',
activation='relu'))
老師您好,因為這地方對於Dense裡的參數不是很理解,因此想請問您裏頭的units和input_dim是甚麼意思,哪裡不相同?
是不是一個指輸入的數目,另一個是指隱藏層的層數呢?謝謝。
回答列表
-
2019/07/15 下午 11:38Jimmy贊同數:1不贊同數:0留言數:3
Hi 渝陽!
沒錯喔! input_dim 指的就是 input dimension 的意思,讓這層 Dense layer 知道你 inputs 的維度是多少,接下來再透過妳給定的 units 來決定 output 的維度囉